#原文的环境是Ubuntu,但是由于没有双系统,安装不了gpu版本的cuda,尝试一段时间后放弃。
因此没有双系统想在VMware中安装gpu的cuda,及早放弃!!
#正文开始:
安装anaconda(安装最新版本的就行),在anaconda里,conda create 环境
查看电脑配置(nvcc-v),安装conda、torch、torchvision,我的电脑显卡是MX250,安装了10.2版本的cuda,10.2版本的cuda toolkit,1.10版本的pytorch,8.3.3版本的cudnn,0.11.1版本的torchvision,3.8版本的python。
注意这几个版本都是需要一一对应,可以上网查一下版本对应关系。(cuda版本和cuda toolkit版本一致;cuda和cudnn版本对应;cuda和pytorch版本对应)
下载到官网去下
之前试过好几次都没能成功下载gpu版本的cuda,好像是cudatoolkit和cudnn都下才行
检验是否下载的是gpu版本:python;import torch;print(torch.__version__)
环境配置成功!!
我仔细参考了一篇csdn播客,但是现在找不到了.....,找到的话我放在评论区。