解决方案:
最终在https://zhuanlan.zhihu.com/p/409616444中找到了答案;
官网提供的conda指令下载的pytorch是CPU版本,换做pip下载即可(记得换源后下载,不然会很慢很慢)
pip换源教程:http://t.csdn.cn/Xd83K
安装过程:
首先我们要保证其它安装是不能出错的(显卡驱动、cuda、cudnn等):
一张图解释显卡驱动、cuda、pytorch版本中的cuda版本关系(来源:我是土堆)
硬件的驱动版本可以向下兼容,因此保证cuda driver version >= cuda runtime version >= pytorch/tensorflow
如果此时你已经安装了错误版本的cuda,可以参考教程卸载;
如果你想安装历史版本的cuda,可以在历史版本中下载runfile格式文件。