这几天在下载pytorch和相关附件的时候掉进了很多的坑,看到论坛上很多的方法是采用了linux系统或者在win10下用anacoda来完成的,但是我又懒得下载anaconda_(:3J∠)_
所以这里写一篇避开anaconda,单纯用pip的win10下载教程吧,希望可以对其他人有帮助。


目录

  • 一、下载torch和torchvision
  • 1.在cmd中用pip下载
  • 2.在官网下载wheel文件
  • 二、下载cuda
  • 三、下载cudnn
  • 四、配置环境
  • 五、检查显卡驱动
  • 六、检查是否能够成功运行
  • 七、资源


一、下载torch和torchvision

1.在cmd中用pip下载

pip install torch torchvision
由于下载速度太慢,该方法一般会失败。

2.在官网下载wheel文件

登录pytorch官网

pip安装pytorch太慢了 pytorch pip install_压缩包


根据需要选择相应的版本,这里选择的是cuda10.1的版本(None是cpu版本),按照图片所示,跳转到https://download.pytorch.org/whl/torch_stable.html

选择需要的wheel文件进行下载

pip安装pytorch太慢了 pytorch pip install_python_02


pip安装pytorch太慢了 pytorch pip install_压缩包_03


因为我的电脑用的是python3.7,所以用的是cp37版

在wheel文件下载成功后,在cmd中pip install wheel文件即可成功安装

但实际过程中即便是下载wheel文件可能也会速率过慢或者下载失败,所以文章最后会给出百度云盘资源

二、下载cuda

登录cuda各个版本链接

pip安装pytorch太慢了 pytorch pip install_压缩包_04


这里选择的是CUDA Toolkit 10.1 update2

下载完成后会得到一个

pip安装pytorch太慢了 pytorch pip install_pip安装pytorch太慢了_05


打开下载,选择自定义下载:

只需选择以下四个(根据自己有没有VS可以选择是否下载Visual Studio Integration):

pip安装pytorch太慢了 pytorch pip install_深度学习_06


下载完毕后关闭即可

三、下载cudnn

登录cudnn下载链接

pip安装pytorch太慢了 pytorch pip install_压缩包_07


点击Login注册并登录,选择cuda对应的cudnn版本进行下载,成功后会得到如下压缩包:

pip安装pytorch太慢了 pytorch pip install_压缩包_08


解压后将文件内的所有内容:

pip安装pytorch太慢了 pytorch pip install_深度学习_09


全部复制到如下文件中(这里是按cuda默认路径下载的结果,如果自定义了cuda的下载路径,请自行找到相应文件夹):

pip安装pytorch太慢了 pytorch pip install_深度学习_10

四、配置环境

打开“我的电脑”属性 → 高级系统设置

pip安装pytorch太慢了 pytorch pip install_深度学习_11


打开环境变量,在系统变量中找到Path:

pip安装pytorch太慢了 pytorch pip install_驱动程序_12


编辑,新建,将C:\Program Files\NVIDIA Corporation\NVSMI添加进环境变量,如图:

pip安装pytorch太慢了 pytorch pip install_pip安装pytorch太慢了_13


若在本步骤中,发现NVIDIA Corporation文件夹中NVSMI文件丢失,解决方法是找一个NVSMI的资源下载并复制到NVIDIA Corporation即可,在本文最后也附上了NVSMI的网盘资源

五、检查显卡驱动

首先查看电脑nvidia显卡的版本,具体步骤如下:

打开“我的电脑”管理 → 设备管理器 → 展开“显示适配器” → 找到相应的nvidia显卡并右键点击 → 属性 → 上方点击“驱动程序”

pip安装pytorch太慢了 pytorch pip install_驱动程序_14


则可以看到自己的显卡配置与驱动程序版本,然后登录驱动程序下载链接

pip安装pytorch太慢了 pytorch pip install_深度学习_15


在左边选择自己的显卡配置,然后在右边“最新”中选择“推荐/认证”,再点击搜索,若自己原有的驱动程序版本不在此列,则下载驱动程序,然后一直默认安装即可

六、检查是否能够成功运行

打开cmd,分别输入命令:nvcc -V 和 nvidia-smi

若出现如下显示则表明环境搭建成功

pip安装pytorch太慢了 pytorch pip install_python_16


然后在Python中检查pytorch是否能够检测到gpu:

In [1]: import torch

In [2]: torch.cuda.is_available()
Out[2]: True

显示True,表明成功检测到可用gpu,然后就大功告成啦

第一次写,如有错漏,请加以指正。

七、资源

网盘中包括的资源有:
1、cuda10.1、python3.7版本的torch-1.4.0
2、cuda10.1、python3.7版本的torchvision-0.5.0
3、cuda10.1对应的cudnn7.6.5.32压缩包
4、补充用的NVSMI压缩包

链接:https://pan.baidu.com/s/1Tfbh-7AAwjCeyZ2YJJqorw 提取码:1uxf