1. 创建一个Pytorch1.1.0的虚拟环境
conda create -n (环境名)(python版本)
conda create -n pytorch1.1 python=3.6
2. 安装 Pytorch
安装方式1:
从官网查看Pytorch1.1.0的安装命令https://pytorch.org/get-started/previous-versions/
# CUDA 10.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
问题:速度太慢,很容易就碰到 httperror.
安装方式2:
自己从清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/里面,下载需要的pytorch版本,然后用命令安装:
conda install pytorch-1.1.0-py3.6_cuda100_cudnn7_1.tar.bz2
发现:pytorch1.1.0都是cuda版本的,没有cpu版本的.
问题:pytorch1.1.0对应的cuda版本跟我电脑上的cuda版本不一样,在以后使用的时候会出现冲突么?
3. 测试Pytorch是否安装成功
import torch
print(torch.cuda.is_available())
输出:True
4. 查看与Pytorch匹配的Cuda版本
import torch;
print(torch.version.cuda)"
输出:10.0
疑问:与Pytorch1.1.0 匹配的 Cuda 版本是10.0,可是我电脑的 Cuda 版本是10.2,所以在以后使用的时候会出现问题嘛?
5. 安装 torch_geometric、torch_sparse、torch_scatter、torch_cluster
(1)依次安装下列包:
pip install --verbose --no-cache-dir torch-scatter
pip install --verbose --no-cache-dir torch-sparse
pip install --verbose --no-cache-dir torch-cluster
pip install --verbose --no-cache-dir torch-spline-conv (optional)
pip install torch_geometric
(2)需要安装指定版本时
- 先从官网找到这几个包的官网历史版本:https://pypi.org/user/rusty1s/,
- 下载.tar.gz 文件到本地,
- 再执行 pip install .tar.gz 文件
(3)当遇到failed to build xxx.whl 这个错误时
1. 到官网找到这几个包的.whl文件,即编译好的各种torch_二进制包:https://github.com/rusty1s/pytorch_sparse
2. 下载你需要的,跟你环境匹配的.whl文件
3. pip install + 刚才下载的.whl 文件的名字:
pip install torch_sparse-0.6.2+cu102-cp37-cp37m-win_amd64.whl
4. 安装成功:
6. 问:与Pytorch1.1匹配的 torch_geometric版本是什么?
上面提到的编译好的二进制文件,都是Pytorch1.4、Pytorch1.5、Pytorch1.6、Pytorch1.7的,并没有pytorch1.1的,所以,如果pytorch1.1装了Pytorch1.4的系列 .whl 文件,也运行的时候可能就会出现问题。
我的出现了问题,但不知道到底是因为什么出了问题,我太难了。。