linux下安装pytorch

名词解释

pytorch是什么？流行的深度学习框架 更详细的介绍请看这里

下载

去pytorch官网根据自己的机器配置选择对应的安装命令

选择好了之后 复制Run this Command 然后到终端执行

需要注意的点

可得会出现下载特别慢的问题

解决方案：将默认的源换成国内的清华源，由于一些原因默认的源国内下载速度过慢

在终端输入以下命令：

# conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# conda config --set show_channel_urls yes

查看修改是否成功

cat ~/.condarc

如果出现下图这个，则修改成功

然后再次执行Run this Command (注意！！！官网上的命令行要把末尾-c pytorch删掉才会用清华源)

检验是否安装成功 在终端顺序执行下面的代码

# python3
# import torch
# import torchvision
# print(torch.__version__)