UserWarning: NVIDIA GeForce RTX 3050 Laptop GPU with CUDA capability sm_86 is not compatible……
摘要:针对NVIDIA RTX 3050等30/40系显卡与PyTorch版本不兼容的问题(显示sm_86架构不支持警告),解决方案为:1)激活conda环境;2)卸载现有PyTorch;3)安装适配CUDA 11.8的版本(使用pip3安装指定whl地址的torch套件)。该方法可解决显卡与深度学习框架的兼容性问题。(98字)
·
关于3050等30、40系显卡不适配torch版本的解决方案:
运行代码后出现以下警告:
UserWarning: NVIDIA GeForce RTX 3050 Laptop GPU 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_61 sm_70 sm_75 compute_37. If you want to use the NVIDIA GeForce RTX 3050 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
解决方案:
①进入环境
conda activate 环境名称
②卸载当前版本
pip uninstall torch torchvision torchaudio -y
③安装适配sm_86的 PyTorch 版本:CUDA 11.8 版本
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
即可正常运行
更多推荐

所有评论(0)