会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
4U
博客园
首页
新随笔
联系
订阅
管理
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
)
收藏
举报
刷新页面
返回顶部
公告