首先安装CUDA,没有安装的就去安装,版本需要比电脑本机的英伟达控制面板相同或者低
然后系统信息
如果说这边版本比较低,例如10.x,我建议卸载更新英伟达控制面板到最新,方法CSDN上很多。
然后去下载CUDA,CUDA下载的版本低于英伟达控制面板支持的即可,我下载的是11.7,下载不低于11.6版本的CUDA。
然后去conda里面创建虚拟环境
conda create -n your_name python=3.8
然后激活虚拟环境,也就是进去到具体虚拟环境中去,这边我也不再赘述
conda activate your_name
然后就是去pyTorch官网看版本查看版本,这边选择pip3,不然可能下载失败,千万不要使用清华镜像(只有CPU)也不要选择conda,因为网很卡,很可能下载失败
然后静静等待下载,下载完成后,验证:
import torch
print(torch.cuda.is_available())
注意:选择pip3的就可以下载,不要用镜像,不要用镜像,不要用镜像,不要用镜像