近几年来,pytorch的发展速度越来越快,在github、CSDN等开源网站上下载的源代码逐渐由tensorflow向pytorch转变。近期在看论文时需要对代码进行复现,在安装pytorch时遇到了很多问题,将整个过程写出来供大家借鉴学习。
注:在进行安装的过程中,通过cmd命令框输入命令发现找不到文件,说明环境变量没有配置好。此方法不需要配置环境变量。

开发环境

python3.x (注意python3.8版本和老版本pycharm不兼容,如果遇到此问题可以更新一下pycharm)
anaconda (个人感觉这个装不装影响应该不大,主要是电脑里面原来就有。而且anaconda很好用,自带python的IDE)
CUDA Toolkit (建议安装10.1版本)

Anaconda安装

1、百度搜索anaconda,进入官网,找到对应的版本下载安装即可。https://www.anaconda.com/products/individual
在这里插入图片描述
2、安装时最好使用默认路径,需要注意的是如果路径中有中文可能导致无法运行。第一个选项勾选之后能够自动将anaconda添加到环境变量中,可以在cmd中输入conda list进行测试。第二个选项勾选之后其他编译软件可以使用anaconda自带的python IDE。
在这里插入图片描述

CUDA 安装

1、百度搜索cuda,进入官网之后找到历史版本信息。https://developer.nvidia.com/cuda-toolkit-archive
这里我选择的是CUDA Toolkit 10.1 (Feb 2019)。
在这里插入图片描述
2、点击按钮之后按照自己电脑的配置选择相应版本,本地使用即可。安装时同样最好选择默认路径。
在这里插入图片描述
3、安装完之后通过cmd命令输入nvcc -V可以看到已经安装完成。
在这里插入图片描述

Pytorch安装

1、打开pytorch官网,https://pytorch.org/get-started/locally/
找到Previous PyTorch Versions,然后找到你想要的安装的pytorch和torchvision版本对应的cmd命令,使用管理员模式运行cmd,输入命令。

在这里插入图片描述

在这里插入图片描述

2、安装完成之后可以打开cmd,输入如下命令进行测试,输出True说明安装完成。
在这里插入图片描述

离线安装PyTorch

如果你需要在服务器或者没有网的机器上安装pytorch,可以参考以下教程。
输入网址:https://download.pytorch.org/whl/torch_stable.html
在这里插入图片描述

找到你要安装的版本,cu100表示cuda版本是10.0,torch-1.0.0表示torch的版本,cp35表示python的版本是3.5,最后部分用于区分linux和windows系统。需要注意的是pytorch版本和torchvision版本需要对应,下载完成之后使用cmd进入文件存放路径,输入pip install 文件名 即可成功安装。

附录

如果你对于上述步骤有任何问题,都可以加QQ1018729230联系博主。

下面是两个参考视频:
https://edu.csdn.net/course/play/16868/245506?spm=1002.2009.3001.4024
https://www.bilibili.com/video/BV1nE411b7Vx?share_medium=android&share_plat=android&share_source=WEIXIN&share_tag=s_i&timestamp=1594875832&unique_k=o20DBu

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐