1、重新创建虚拟环境
2、查找pytorch版本
安装Pytorch如何选择CUDA的版本,看这一篇就够了 - 知乎 (zhihu.com)
Previous PyTorch Versions | PyTorch
安装支持CUDA 12的pytorch教程 - 知乎 (zhihu.com)
于是:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
(未开net,因为昨天开着报错了,真的好慢,还怕清华源变成CPU的...)
完成
试了一下啊,cuda可用。
标签:知乎,PyTorch,pytorch,12.0,cuda,安装 From: https://www.cnblogs.com/hxpp/p/17810830.html