更新显卡驱动可以安装高版本的cuda
可以通过打开cmd输入nvidia-smi来查看

所以说不高于12.3的版本都可以安装
cuda版本匹配pytorch版本
打开网址:https://pytorch.org/

验证完后,就可以吭哧吭哧的去下载安装了
搞完后还是发型代码跑出来GPU = 0
不要慌
关键一步


把这个复制进去安装一下
运行

根本原因
pycharm安装的torch只支持cup计算,所以需要重新装过

本文介绍了如何更新显卡驱动以便安装高版本CUDA,强调了CUDA版本需与PyTorch版本匹配。特别指出在PyCharm中使用PyTorch时,可能需要重新安装以支持cup计算。最后提供了关键步骤和解决GPU显示问题的方法。
可以通过打开cmd输入nvidia-smi来查看

所以说不高于12.3的版本都可以安装
打开网址:https://pytorch.org/

验证完后,就可以吭哧吭哧的去下载安装了
搞完后还是发型代码跑出来GPU = 0
不要慌


把这个复制进去安装一下

pycharm安装的torch只支持cup计算,所以需要重新装过

您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
3066
2304
4388