安装cuda 、torch、torchvision

整体思路为 先根据 nvidia-smi 指令 看看自己的cuda版本,然后安装对应的cuda驱动 对应关系表如下:

cuda版本号查询

接着,根据安装的cuda以及python版本,安装对应的torch,cuda、python与torch对应版本号如下:

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
Albart5756 天前
【已解决】ModuleNotFoundError_ No module named ‘xxx’ 模块导入失败终极解决
python·pip·环境配置·后端开发·bug解决·模块报错·python导包失败
2401_853448236 天前
YOLO开发环境搭建
yolo·环境配置·ultralytics
久久学姐15 天前
Python安装与环境配置全程详细教学(包含Windows版和Mac版)
环境配置·mac版·python安装·windows版·ide使用
脱脱克克2 个月前
使用 TRAE / VS Code + DeepSeek V4 开发微信小程序、网页
微信小程序·ai编程·环境配置
charlee443 个月前
《GIS基础原理与技术实践》配套案例(Python版)
python·conda·numpy·gis·环境配置
Thanks_ks3 个月前
【第 002 讲】Python 标准开发环境搭建:运行环境 | 环境变量 | IDE 部署 | 配置优化
ide·python·pycharm·开发工具·环境配置·环境变量·编程基础
SkyXZ3 个月前
Mac上使用VScode优雅开发STM32
环境配置
不会写程序的未来程序员4 个月前
nvm 安装教程:Node.js 版本管理全攻略 (Win/Mac/Linux) + .nvmrc 实战
linux·macos·node.js·前端开发·环境配置·nvm
陈嘿萌5 个月前
4090显卡配置官方SegFormer运行环境(包含预训练权重下载、环境配置、代码修改、运行 | 验证 | 测试)
环境配置·模型训练·segfomer·预训练权重下载