Pytorch1.1.0 + cuda10.0+torch_geometric

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)需要安装指定版本时
  1. 先从官网找到这几个包的官网历史版本:https://pypi.org/user/rusty1s/
  2. 下载.tar.gz 文件到本地,
  3. 再执行 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 文件,也运行的时候可能就会出现问题。

我的出现了问题,但不知道到底是因为什么出了问题,我太难了。。

在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值