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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM