【基建笔记】通过本地文件安装 pytorch 和 torchvision

通过本地文件安装pytorch1.4.0和torchvision0.5.0

0.说明:

1. 用Pytorch官网中提供的安装命令真的巨慢,
2. 即便在conda中添加了清华源路径还是如此. 具体操作可见 参考博客
3. 所以直接在清华源找到对应的版本文件,通过浏览器下载后再手动安装. https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
4. 以下基于anaconda平台的conda指令进行安装, pip暂未实验.

1.下载合适版本的pytorch与torchvision

1. pytorch和torchvision是分开的,如果不清除可以看下pytoch官网的安装命令.
2. 注意需要和CUDA版本对应上. 我电脑里装的是CUDA10.0.
3. 所以下载的文件分别是pytorch-1.4.0-py3.6_cuda10.0.130_cudnn7.6.3_0.tar.bz2和torchvision-0.5.0-py36_cu100.tar.bz2.下载速度飞快.

2. 安装

1. conda切换到环境下.
2. 在下载目录下可通过conda直接对压缩包进行安装,解压安装的速度是很快的.

conda install pytorch-1.4.0-py3.6_cuda10.0.130_cudnn7.6.3_0.tar.bz2
conda install torchvision-0.5.0-py36_cu100.tar.bz2

3. 把上面的conda换成pip就是pip的安装方法,
4. 还可以通过命令指定位置来进行安装:

pip install --target=......

3.验证:

1. 在终端python验证:

import torch 
print (torch.__version__)

2. 如果出现问题,一般是CUDA的问题.仔细简单CUDA的环境配置和pytorch \ torchvision 版本是否对应.