安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
王伯安呢7 天前
Python 环境配置初学者指南:从安装到 Pycharm 项目配置
开发语言·python·pycharm·环境配置·初学者
马甲是掉不了一点的<.<25 天前
Stable Diffusion 环境配置详细指南
stable diffusion·环境配置
旧时光巷1 个月前
【docker①】在VS Code中使用Docker容器
运维·vscode·docker·容器·环境配置·安装教程·镜像构建
handsomestWei1 个月前
Cursor Java开发配置
java·ai编程·环境配置·cursor
seasonsyy1 个月前
2.安装CUDA详细步骤(含安装截图)
python·深度学习·环境配置·cuda
seasonsyy1 个月前
1.安装anaconda详细步骤(含安装截图)
python·深度学习·环境配置
mahuifa3 个月前
PySide环境配置及工具使用
python·qt·环境配置·开发经验·pyside
瓢儿菜20183 个月前
Proteus8.17-LCD12864液晶屏幕仿真模型
单片机·proteus·环境配置·仿真
妄想成为master4 个月前
如何完美安装GPU版本的torch、torchvision----解决torch安装慢 无法安装 需要翻墙安装 安装的是GPU版本但无法使用的GPU的错误
人工智能·pytorch·python·环境配置
瓢儿菜20185 个月前
proteus8.17 环境配置
单片机·proteus·环境配置