安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
SkyXZ11 小时前
告别虚拟机!WSL2安装配置教程!!!
环境配置
SkyXZ4 天前
老奶奶看了都会的WSL2连接USB设备教程!
环境配置
卡戎-caryon18 天前
【MySQL】01.MySQL环境安装
数据库·笔记·mysql·环境配置
SkyXZ20 天前
2025年了,你还不会配置Jetson Orin NX嘛?
环境配置·jetson
程序猿000001号22 天前
Conda 使用全解析:从入门到精通
conda·环境配置
SpikeKing2 个月前
LLM - 使用 LLaMA-Factory 微调大模型 环境配置与训练推理 教程 (1)
人工智能·llm·大语言模型·llama·环境配置·llamafactory·训练框架
炼丹师小米3 个月前
Ubuntu24.04.1系统下VideoMamba环境配置
python·环境配置·videomamba
坚果仙人4 个月前
【环境配置】AST: Asymmetric Student-Teacher Networks for Industrial Anomaly Detection
环境配置
QH_ShareHub4 个月前
解决R包依赖版本不兼容问题
r语言·环境配置·r包