1. 程式人生 > 其它 >windows下cuda、cudnn以及pytorch的安裝

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命令