问题描述
前不久给新来的 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 及以上版本。
引用参考
文章出处登录后可见！
已经登录？立即刷新
