cuda+pytorch环境安装
1. 安装cuda
nvidia-smi
查看GPU信息,显示的CUDA Version是可以支持的最高版本cuda- 去官网安装相应版本的cuda
- 安装后重新打开终端,
nvcc --version
,输出cud版本
2. 安装conda/miniconda
2.1 安装
到anaconda官网按指示安装,使用onda --version
看conda版本
- macos
- win 下载exe安装
2.2 常用conda命令
2.2.1 conda配置信息
conda info
查看conda信息,里面有配置文件位置.condarc
conda config --show [xx]
显示某项- 一般配置一些channel和proxy
conda config --show channels
conda config --add channels conda-forge
conda config --remove channels conda-forge
- conda镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
2.2.2 环境管理
conda create --name xxx python=3.12
# 移除
conda remove --name xxx --all
conda remove --name xxx package_name
# 导出
conda env export --name xxx > xxx.yml
# 还原
conda env create -f xxx.yml
2.2.3 conda包管理
conda list # 查看当前环境中的包
conda list package_name # 查看当前环境中是否安装了
conda list package_name* #模糊查询
conda search package_name # 在anaconda repository中查询目标包
conda update package_name
2.2.4 conda install & pip install
- 先conda再pip
- pip是给指定的环境安装的
3. 安装PyTorch
nvcc -V
查看cuda版本
torch官网
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch