腾讯云+python3.7+pytorch+cuda10.0+cudnn+ubuntu18.04

不为别的,就为了写给自己,我的记忆只有三秒,再多一秒都不行

1. 首先上腾讯云,开账号,选服务,我选的是这个服务,注意安装的时候要选镜像,我不想装cuda了,我累了,所以直接选装好的。不要贪便宜选3块多的,因为你会连这三块钱也浪费了,因为这个虽然是写NVIDIA T4,实际上我显示显卡的时候是curixxx,不是NVIDIA,我不知道怎么弄,所以还是买个6块钱的吧。腾讯云+python3.7+pytorch+cuda10.0+cudnn+ubuntu18.04_第1张图片

腾讯云+python3.7+pytorch+cuda10.0+cudnn+ubuntu18.04_第2张图片

腾讯云+python3.7+pytorch+cuda10.0+cudnn+ubuntu18.04_第3张图片

2. 安装python3.7,其实它自带2.7,但我不想要,我的代码本身就在3.7跑的,我坚决换回3.7。安装就是apt来一套

3. 安装完3.7后,输入https://pytorch.org/get-started/previous-versions/

python3.7 -m pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html 

请不要直接用pytorch现在官网给的只有10.1cuda的,我们镜像带的cuda是10.0的,经验告诉我,比你自己带的高的torch都不行,按照这句代码其实得出来torch.cuda的版本是9.2.148,但是cuda是能用的。

4. 这部之后就完成了,是的,就是这么简单,输入

print(torch.cuda.is_available())

返回的是True

 

 

题外话:一直以为自己的是因为驱动器的版本有问题才导致一直不行,但是我自己上nvidia的官网对照服务器的版本下载的就是410版本,并且本身镜像自带的就是410版本的。说到底就是走了个弯路。总结一句:钱真是个好东西

你可能感兴趣的:(Linux)