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

发布时间 2023-11-05 18:43:54作者: 春山可

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