有关NVIDIA 4090的pytorch版本安装心得

目录

1.查看版本

输入 nvidia-smi 查看GPU支持的cuda最高版本

如上图所示,你的GPU支持的最高CUDA版本便为12.2,所以你要安装CUDA版本不能高于该版本。安装版本不能太低,建议就小一个版本就可以。

nvcc -V查看你安装的CUDA版本

如上图所示,我安装的CUDA版本为:12.1

2.安装对应版本的pytorch

定义: pytorch 中的CUDA版本不能高于当前机器上已经安装的CUDA版本(12.1)

故我安装的pytorch对应的版本里面的CUDA版本为12.1

安装链接:在该链接内找到对应的pytorch版本,建议使用pip安装。

3.检测是否安装成功

  • conda activate name ( 激活相关的虚拟环境)

  • 输入python 进入python 环境

  • 输入 import torch 导入torch 安装包

  • 测试 torch.cuda.is_available() ,返回True则成功

  • 测试 torch.zeros(1).cuda() ,返回如下图所示将结果成功

4.问题

这里面会报各种各样的问题,总之都是跟版本不兼容有关,但里面有一个错误比较奇怪,如下图所示。

复制代码
/home/respecting/anaconda3/envs/torch1.8.1/lib/python3.7/site-packages/torch/cuda/__init__.py:104: UserWarning: 
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

当你步骤3的检测全部成功后,在跑你的人工智能模型时依旧会报该错误,说明虽然版本选对了,可能太新了,建议将pytorch的版本降低一些。

相关推荐
思则变1 小时前
[Pytest] [Part 2]增加 log功能
开发语言·python·pytest
叶子爱分享1 小时前
计算机视觉与图像处理的关系
图像处理·人工智能·计算机视觉
鱼摆摆拜拜1 小时前
第 3 章:神经网络如何学习
人工智能·神经网络·学习
一只鹿鹿鹿1 小时前
信息化项目验收,软件工程评审和检查表单
大数据·人工智能·后端·智慧城市·软件工程
张较瘦_1 小时前
[论文阅读] 人工智能 | 深度学习系统崩溃恢复新方案:DaiFu框架的原位修复技术
论文阅读·人工智能·深度学习
cver1231 小时前
野生动物检测数据集介绍-5,138张图片 野生动物保护监测 智能狩猎相机系统 生态研究与调查
人工智能·pytorch·深度学习·目标检测·计算机视觉·目标跟踪
漫谈网络1 小时前
WebSocket 在前后端的完整使用流程
javascript·python·websocket
学技术的大胜嗷1 小时前
离线迁移 Conda 环境到 Windows 服务器:用 conda-pack 摆脱硬路径限制
人工智能·深度学习·yolo·目标检测·机器学习
还有糕手1 小时前
西南交通大学【机器学习实验10】
人工智能·机器学习
江瀚视野2 小时前
百度文心大模型4.5系列正式开源,开源会给百度带来什么?
人工智能