不为别的,就为了写给自己,我的记忆只有三秒,再多一秒都不行
1. 首先上腾讯云,开账号,选服务,我选的是这个服务,注意安装的时候要选镜像,我不想装cuda了,我累了,所以直接选装好的。不要贪便宜选3块多的,因为你会连这三块钱也浪费了,因为这个虽然是写NVIDIA T4,实际上我显示显卡的时候是curixxx,不是NVIDIA,我不知道怎么弄,所以还是买个6块钱的吧。
2. 安装python3.7,其实它自带2.7,但我不想要,我的代码本身就在3.7跑的,我坚决换回3.7。安装就是apt来一套
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版本的。说到底就是走了个弯路。总结一句:钱真是个好东西