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