Pytorch安装教程
Pytorch安装教程开发环境Anaconda安装CUDA 安装Pytorch安装附录近几年来,pytorch的发展速度越来越快,在github、CSDN等开源网站上下载的源代码逐渐由tensorflow向pytorch转变。近期在看论文时需要对代码进行复现,在安装pytorch时遇到了很多问题,将整个过程写出来供大家借鉴学习。注:在进行安装的过程中,通过cmd命令框输入命令发现找不到文件,说明环
Pytorch安装教程
近几年来,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×tamp=1594875832&unique_k=o20DBu
更多推荐
所有评论(0)