cuda安装

1、查看主机GPU支持的最高cuda版本

cmd命令 nvidia-smi

红框中为GPU支持的cuda最高版本

2、根据Pytorch选择对应的cuda版本

在Pytorch官网查看pytorch对应的cuda版本(pytorch官网

例如

Pytorch2.5对应的cuda版本有3个:11.8、12.1、12.4,但是我们只能选择11.8和12.1,因为第一步中我们知道GPU支持的cuda最高版本为12.1。

3、下载对应版本的cuda(cuda下载链接

4、查看cuda安装成功

cmd命令 nvcc -V

posted @ 2024-11-12 13:36  4U  阅读(104)  评论(0)    收藏  举报