一.在anaconda下配置cuda、cudnn以及pytorch環境
1.打開Anaconda Prompt,輸入
conda create -n pytorch python=3.8 (這里環境名字命名為pytorch)
2.激活環境
activate pytorch
3.查看自己電腦顯卡驅動版本,以及顯卡驅動對應的最高cuda版本
在dos命令行輸入
nvidia-smi
本機顯卡驅動為511.79,如果不是最新版本可以先把驅動升級為最新版本,因為驅動對cuda的版本是向下兼容的,這樣可以選擇的cuda版本會更多
本機兼容cuda最高版本為11.6
4.進入pytorch官網,找到合適的cuda版本對應的pytorch,復制安裝的代碼到激活的pytorch環境中
在pytorch環境輸入conda install pytorch torchvision torchaudio cudatoolkit=11.3 ,等待安裝即可
二.在windows下配置cuda、cudnn以及pytorch環境
1.下載Python解釋器
到Python官網下載Python解釋器,本文下載Python3.8版本,下載下來后安裝即可
2.根據一種的方法到NVIDIA官網下載對應cuda版本
3.到NVIDIA官網下載cuda版本對應的cudnn,並把文件復制到cuda制定安裝目錄中
4.配置pytorch
進入pytorch官網,如一所示,package選擇pip,復制對應代碼到dos窗口,如果下載出現時間過長,安裝失敗錯誤
可用pip --default-timeout=100 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html命令