windows下cuda、cudnn以及pytorch的安裝
一.在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命令