降torch 版本到1.2
1、使用conda 进行降版本,超级简单,不需要降低cuda 版本,具体操作,以我的cuda 10.0 为例子,查阅网上都说1.2 torch 不支持cuda 10.0,我还真信了。
2、查阅自己的cuda 版本 cmd 输入 nvcc -V
nvcc -V
3、查阅自己的python 版本 ,conda 激活空间输入(注意不是cmd)python --version
4、登录到清华镜像寻找torch1.2 对应的cuda 版本
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
进入后惊奇的发现,有cuda10 适配的torch 1.2版本,不用降低cuda 版本,美滋滋。
5、下载下来,然后使用conda 直接安装,他会默认替换掉你以前的版本,在激活空(不是cmd)输入
conda install --use-local E:\anconda\pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2
# 注意后面为你下载文件放在本地的路径
但是暂时没看见高于cuda 10的版本的,可以去阿里镜像看看有没有,这里我懒得看了。
6、验证是否安装成功
>>> import torch
>>> print(torch.__version__,torch.cuda.is_available())
参考博客
1、https://www.pythonheidong.com/blog/article/468211/4004e62aa2e1fc5f19fe/