win10 pytorch安装过程
查看一些重要参数
参数名称 | 查看方法 |
---|---|
CUDA版本查看 | 如果已经安装过CUDA可以通过命令nvcc -V 可以查看当前CUDA的版本 |
cuDNN版本查看 | 已经安装过cuDNN可以在默认安装位置C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\inculde\bin\cudnn.h 用记事本打开,可以看到cuDNN的版本(好像旧版本的可以,新版本的看不到了) |
显卡型号查看 | 我的电脑->属性->设备管理器->显示适配器 查看 |
查看NVIDIA驱动对应的CUDA版本 | 右击鼠标NVIDIA控制面板->左下角系统信息->组件->可以查看对应CUDA版本 |
坎坷经历
曾经安装过NVIDIA驱动,可能因为windows自动更新导致不可用,一开始以为NVIDIA、CUDA、cuDNN这些都有,就直接去安装pytorch了……结果我的NVIDIA控制面板都打不开
所以先检查一下自己的NVIDIA可不可用!!
安装NVIDIA驱动
可以卸载掉所有的NVIDIA驱动重新安装
卸载NVIDIA
- 控制面板->卸载程序->所有带NVIDIA的全部卸载(有些可能会要求卸载完立即重启,就多重启几次,不要怕耽误时间)
- 卸载完后在
C:\Program Files
和C:\Program Files(x86)
中所有带NVIDIA的文件夹全部删掉
安装NVIDIA驱动
先查看自己的显卡型号,然后到NVIDIA官网输入自己的显卡型号,搜索对应的NVIDIA驱动下载(我就直接下载了第一个),下载完傻瓜式安装就可以,建议不要更改默认路径
安装CUDA
查找NVIDIA对应的CUDA版本,CUDA官网下载,这里是11.1版本的,之前的版本可能要自己找一下,下载完后同样傻瓜式安装
安装cuDNN
同样官网下载CUDA版本对应的cuDNN
注意看清对应的CUDA版本再下载
下载完后是一个压缩包,解压之后,分别将压缩包内的bin、include、lib\x64中的文件,复制粘贴到CUDA路径(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
)下对应的同样名称的文件夹里即可
安装pytorch
首先使用anaconda建一个python36的虚拟环境 conda create -n pytorch python=3.6
然后 activate pytorch
去pytorch官网中选择你操作系统、CUDA版本等、最下方会出来安装命令
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
复制然后切换回cmd粘贴命令安装即可
检查是否成功安装
import torch
不报错说明可以在cpu下使用pytorch
torch.cuda.is_availabel()
返回True说明可以在GPU下使用pytorch
至此安装就结束了~~