打开pytorch安装网址,根据系统配置选择特定版本
在Terminal输入命令,查看CUDA Version
nvidia-smi
e.g.我的CUDA版本是11.2,下载CUDA11.8可向下兼容,因此对应我的操作系统这样选择
复制下面 Command 到 Terminal 中运行
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
完成后运行以下代码进行测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())
成功后最后一个会显示true
注意!
一开始其实是false,于是conda list了一下发现,pytorch-mutex是cpu版本,但是不管如何uninstall还是按照其他方法重下,重安再卸cpuonly,还是不行。索性把环境删了冲装,这次我装的python版本是3.9(上一次是3.7),于是奇迹般地True了!