Pytorch (1) Install

Pytorch Install

Pytorch (1) Install

安装之前

由于之前安装pytorch碰到很多问题,这里进行汇总,希望对大家有所帮助

本人的电脑环境时:win10,python3.6,cuda9.0
首先为pytorch新建一个anaconda环境,在命令行中输入

conda create --name mytorch python=3.6 # 创建一个python3.6的环境,我们把它命名为mytorch

现在我们再用conda info --env或conda env list查看当前所有环境,出现我们的新建的环境即成功

conda env list

然后使用conda activate mytorch 来**刚才创建的环境

conda activate mytorch

在线安装

直接登录pytorch官网,网址:https://pytorch.org/ ,按照图示选择配置,在命令行中输入图中命令
Pytorch (1) Install
但是,本人在安装过程中下载巨慢,中途还会自动中断,更换了清华和中科大的源都不行,最终放弃

下面是换源的过程,不会的可以看看,很简单
https://mirrors.ustc.edu.cn/anaconda/pkgs/free/ 是中科大的源对应的地址,其他的自己去找就行

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

然后查看当前所有源

conda config --show channels

其实所有的源都保存在用户文件夹下的. condarc文件中,直接去修改也行

离线安装

顾明需要下载相关的软件包,这里去清华的镜像站下载所需的.tar.bz2,

本人选择的是 2019.2.18更新的 pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2
然后在文件所在目录下,输入

conda install --offline -n XXXX..tar.bz2
conda install torchvision

最后输入以下命令,测试

import torch
print(torch.__version__)

输出版本号: 1.0.1 即成功

可能出现问题

若出现找不到模块命令,则很可能是环境变量配置问题,找不到torch_python.lib,需要添加环境路径
路径为:C:\Users\surname\Miniconda3\envs\mytorch\Lib\site-packages\torch\lib

参考链接

1、https://blog.csdn.net/Suan2014/article/details/80410144
2、https://blog.csdn.net/lijiecong/article/details/78339078