安装pytorch,首先需要确认你的电脑是否有可支持CUDA的显卡(GPU),不然最后安装会显示False
【像我下面一样,因为我的GPU是AMD,兼容不了CUDA,所以没成功。下面是本人没有确认自己显卡是否能够支持CUDA导致的False,在这里只是记下来以便后续需要,仅供参考,若有错误还请指出,以免误导大家,谢谢支持。】
在笔记本下方任务栏,点击鼠标右键-->任务管理器-->GPU(可查看自己GPU型号)
然后根据GPU型号查看CUDA相应的版本:
1.可根据下面这个网站看:
https://www.geforce.cn/hardware/technology/cuda/supported-gpus
2.也可以参考下面:
Tesla 系列: 用于技术和科学计算,例如 Tesla K80、Tesla K40、Tesla K20 等,具有不同的计算能力,如 3.5、3.7 。 Quadro 系列: 用于专业可视化,例如 Quadro RTX 8000、Quadro RTX 5000、Quadro P6000 等,具有 7.5、6.1 等计算能力 。 GeForce 系列 和TITAN 系列: 用于游戏娱乐,例如 GeForce RTX 3090、GeForce RTX 3080、NVIDIA TITAN RTX 等,具有 8.6、7.5 等计算能力 。 Jetson 系列: 用于边缘AI自主机器,虽然搜索结果中没有列出具体的Jetson产品,但它们也是支持CUDA的 。
个人根据自己学习的情况总结,仅供参考