升级显卡驱动，下载cuda和cudann 12.0后，安装pytorch

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可用。