【PyTorch】切记：GeForce RTX 3090 显卡仅支持 CUDA 11 以上的版本！

问题描述

前不久给新来的 2台 8 张 GeForce RTX 3090 服务器配置了深度学习环境（配置教程参考这篇文章），最近在使用的时候却遇到了各种问题。

问题 1：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 GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

问题 2：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

解决方案

原来是 GeForce RTX 3090 显卡仅支持 CUDA 11 以上的版本！而从官网得知 PyTorch 1.7.0 开始才支持 CUDA 11，所以要使用 GPU 训练的话，必须安装 PyTorch 1.7.0 及以上版本。

引用参考

CUDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方法