win10 pytorch安装过程

win10 pytorch安装过程

  • 查看一些重要参数
  • 坎坷经历
  • 安装NVIDIA驱动
    • 卸载NVIDIA
    • 安装NVIDIA驱动
  • 安装CUDA
  • 安装cuDNN
  • 安装pytorch
  • 检查是否成功安装

查看一些重要参数

参数名称 查看方法
CUDA版本查看 如果已经安装过CUDA可以通过命令nvcc -V 可以查看当前CUDA的版本
cuDNN版本查看 已经安装过cuDNN可以在默认安装位置C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\inculde\bin\cudnn.h 用记事本打开,可以看到cuDNN的版本(好像旧版本的可以,新版本的看不到了)
显卡型号查看 我的电脑->属性->设备管理器->显示适配器查看win10 pytorch安装过程_第1张图片
查看NVIDIA驱动对应的CUDA版本 右击鼠标NVIDIA控制面板->左下角系统信息->组件->可以查看对应CUDA版本win10 pytorch安装过程_第2张图片

坎坷经历

曾经安装过NVIDIA驱动,可能因为windows自动更新导致不可用,一开始以为NVIDIA、CUDA、cuDNN这些都有,就直接去安装pytorch了……结果我的NVIDIA控制面板都打不开
所以先检查一下自己的NVIDIA可不可用!!

安装NVIDIA驱动

可以卸载掉所有的NVIDIA驱动重新安装

卸载NVIDIA

  1. 控制面板->卸载程序->所有带NVIDIA的全部卸载(有些可能会要求卸载完立即重启,就多重启几次,不要怕耽误时间)
  2. 卸载完后在C:\Program FilesC:\Program Files(x86)中所有带NVIDIA的文件夹全部删掉

安装NVIDIA驱动

先查看自己的显卡型号,然后到NVIDIA官网输入自己的显卡型号,搜索对应的NVIDIA驱动下载(我就直接下载了第一个),下载完傻瓜式安装就可以,建议不要更改默认路径
win10 pytorch安装过程_第3张图片
win10 pytorch安装过程_第4张图片

安装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
win10 pytorch安装过程_第5张图片

复制然后切换回cmd粘贴命令安装即可

检查是否成功安装

import torch不报错说明可以在cpu下使用pytorch
torch.cuda.is_availabel()返回True说明可以在GPU下使用pytorch

win10 pytorch安装过程_第6张图片

至此安装就结束了~~

你可能感兴趣的:(pytorch)