『PyTorch』安装教程

最新推荐文章于 2024-07-12 09:59:57 发布

ツぃ☆ve芜情

最新推荐文章于 2024-07-12 09:59:57 发布

阅读量4.2k

点赞数 2

分类专栏： PyTorch 文章标签： python pytorch anaconda

本文链接：https://blog.csdn.net/dreaming_coder/article/details/104307727

版权

PyTorch 专栏收录该内容

21 篇文章 3 订阅

订阅专栏

1. 安装Anaconda

下载地址：https://www.anaconda.com/，傻瓜式安装，记得勾选【add to path】
要学会创建虚拟环境，这个百度一大把

2. 安装pycharm

下载地址：https://www.jetbrains.com/pycharm/，基本也是傻瓜式安装，也要记得勾选【add to path】
这里可以安装社区版，专业版你要能找到破解方法破解当然更好

3. pytorch安装(GPU版本)

一般国内直接安装太慢，这里使用离线安装。

python版本：3.6
cuda版本：9.2
pytorch版本：1.4.0
torchvision版本：0.5.0

3.1 下载cuda安装包

点击cuda9.2进入下载页面

根据自己的系统选择
开始下载安装包

3.2 下载cudnn安装包

点击cudnn进入下载页面(此下载必须登录账号，没有就先注册)
开始下载
选择相对应cuda版本下载
选择对应操作系统的版本

3.3 安装cuda

Step 1
运行安装包，选择安装路径
Step 2
选择【精简】模式，点击【下一步】
Step 3
勾选【接受】，点击【NEXT】
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xNFd7mJG-1581647317308)(E:\我的学习\pytorch\figure\figure7.png)]$
Step 4
点击【下一步】
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-2neJCsnQ-1581647317310)(E:\我的学习\pytorch\figure\figure8.png)]$
Step 5
取消勾选【创建桌面快捷方式】，点击【关闭】
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YnQotbza-1581647317314)(E:\我的学习\pytorch\figure\figure9.png)]$
Step 6
验证cuda是否安装成功
按住Win+E，输入【cmd】，打开命令行，输入
```
nvcc -V
```
有如下返回则安装成功：
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-e3XImI4N-1581647317316)(E:\我的学习\pytorch\figure\figure10.png)]$

3.4 安装cudnn

Step 1
解压压缩包，压缩包内文件如下：
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-BGoECp2P-1581647317319)(E:\我的学习\pytorch\figure\figure11.png)]$
Step 2
打开cuda安装文件夹，默认的强制安装路径为：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

之前选择的路径是解压路径，安装完成默认删除，打开安装路径如下：
Step 3
将解压的cudnn压缩包内的三个文件夹复制到cuda安装目录下：
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-eWPstwvH-1581647317324)(E:\我的学习\pytorch\figure\figure13.png)]$
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-uUgek5iP-1581647317325)(E:\我的学习\pytorch\figure\figure14.png)]$
Step 4
验证cudnn是否安装完成按住Win+E，输入【cmd】，打开命令行，输入
```
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite
```
进入demo文件夹，然后执行两个执行程序：
```
bandwidthTest.exe
```
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-H9HgSknU-1581647317328)(E:\我的学习\pytorch\figure\figure15.png)]$
```
deviceQuery.exe
```
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-DpqZaD5b-1581647317331)(E:\我的学习\pytorch\figure\figure16.png)]$

3.5 下载pytorch离线文件

进入pytorch官网：https://pytorch.org/，点击【Get Started】

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SJwkgM94-1581647317335)(E:\我的学习\pytorch\figure\figure17.png)]$
选择适合电脑的配置

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-kz38fEcX-1581647317338)(E:\我的学习\pytorch\figure\figure18.png)]$

记住勾选【pip】，这样才会出现whl文件的网址
打开whl文件网址
因为直接运行此代码太慢了，可以复制其中的网址，打开后选择适合自己的版本下载

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xPNFGvvk-1581647317340)(E:\我的学习\pytorch\figure\figure19.png)]$

网址打开后如下所示：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-tl5seX8x-1581647317341)(E:\我的学习\pytorch\figure\figure20.png)]$

此处文件的命名时有规律的：

选择需要的版本下载，可以用Ctrl+F搜索，找到所需文件单击下载

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-eFBHba9k-1581647317347)(E:\我的学习\pytorch\figure\figure22.png)]$

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-MdlHUqiV-1581647317350)(E:\我的学习\pytorch\figure\figure23.png)]$

3.6 安装pytorch

打开pycharm，新建project
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-tszdzhIC-1581647317353)(E:\我的学习\pytorch\figure\figure24.png)]$
这里要选择【Existing interpreter】，选择你创建的anaconda虚拟环境，第一次创建的话，可能不会直接显示，你需要自己添加虚拟环境的解释器：
创建python file，写入以下内容
```
import torch

print(f"Hello PyTorch {torch.__version__}")
```
运行后会显示以下错误：
ModuleNotFoundError: No module named 'torch'`
下面就开始安装所缺的包
打开pycharm的【Terminal】窗口
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-vH5tjj3F-1581647317355)(E:\我的学习\pytorch\figure\figure28.png)]$

输入 cd 命令，进入前面所下载的离线文件的路径：
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-2OnD0eDE-1581647317356)(E:\我的学习\pytorch\figure\figure29.png)]$
安装torch

输入pip install torch，按住Tab键，即可自动补全：
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8dPQZQyg-1581647317360)(E:\我的学习\pytorch\figure\figure30.png)]$
回车进行安装
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-pa4XDuSZ-1581647317363)(E:\我的学习\pytorch\figure\figure31.png)]$

下面类似方法安装torchvision：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-H8zPX836-1581647317373)(E:\我的学习\pytorch\figure\figure32.png)]$
再次运行代码，可以发现正常运行
$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-o26vhpVg-1581647317375)(E:\我的学习\pytorch\figure\figure33.png)]$
OK，至此安装完成.