前言
谁知道我以后会不会再掉坑。所以记一下
技巧
1、如何查看py版本
打开cmd-》输入python
我的显示:
Python 3.7.3 (v3.7.3:ef4ec6ed12, Mar 25 2019, 22:22:05) [MSC v.1916 64 bit (AMD64)] on win32
Type “help”, “copyright”, “credits” or “license” for more information.
2、清华镜像源
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
3、pytorch的官网
https://pytorch.org/get-started/locally/
4、如何查看自己电脑的cuda?
(1)找到这个东西
(2)打开
(3)点
5、如何给自己的垃圾cuda驱动升个级?
如果我没记错,应该是下图,右键选择升级驱动即可
开始之前
注意到,在cuda小版本更新下差别不大。例如本机驱动为11.4版本,可以使用11.3的pytorch对应的cuda版本。
但跨大版本会导致变量、模型在gpu、cpu之间的复制产生问题,从而导致不可用。
因此,最简单的方式还是按照官网找到对应的版本相接近的指令直接pip
如果无法满足需求可以继续看下边的坑。
一个萝卜一个坑
坑1
不能直接pip,找不到对应版本号,等死吧。
坑2
就算你知道要去官网,找到对应版本号。可能会遇到很多问题
例如cuda版本不对,可以尝试下载老版本。最后选择了升级了一下显卡驱动(见上)。
然后开始下载!
失败-超时。
坑3
好了,接下来自然想着换镜像源
失败,尝试换了好几个国内镜像都失败。
嗯,不错,我很开心。
坑4
接下里,尝试离线下载。
博主打算下载这个。
复制一下后边那个html那个网页
打开,界面如下
博主通过推理数字代表的意义,发现第一个就是你的cuda版本 第二个是你要下的torch(或torchvision)版本号,第三个是对应的你的py版本号,然后是选择linux或者windows。
启动。…
问题不大:多试几遍就能下载下来。博主用的IDM。
坑5
当用上述方法,安装完两个库之后。
cmd 操作:cd找到对应安装位置。
然后pip install 名字.whl;这两个文件就都被安装了
中间可能会因为顺序原因,报错
eg.
ERROR: torchvision 0.5.0 has requirement torch==1.4.0, but you’ll have torch 1.5.0+cu101 which is incompatible.
问题不大,两个包按顺序各按几次。
最最最后一个小问题
直接 pip install wheel解决即可。
最后做一下测试
import torch
import torchvision
print(torch.cuda.is_available())
总结
1、打开官网找到对应版本
2、对着版本打开下面代码部分的网页,然后按照上述推测,下载对应你系统的版本号(一定要提前看好)
3、cd到下载位置,安装两个包,顺道安装wheel这个包。
后记
一壶茶一包烟,一个环境陪一天。
老老实实上课多好,整什么这些东西。