遇到问题

NVIDIA GeForce RTX 3090 Ti with CUDA capability sm_86 is not compatible with the current PyTorch..._官网

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

查看显卡型号算力

官网网址:​​https://developer.nvidia.com/cuda-gpus​

NVIDIA GeForce RTX 3090 Ti with CUDA capability sm_86 is not compatible with the current PyTorch..._环境配置_02

使用的RTX 3090的算力是8.6,cuda支持的算力应当要高于显卡的算力。

不仅有torch版本,torch还包含有对应的cuda版本,RTX30系列的显卡算力比较高,对应的cuda版本不能是10.x,只能是11.x. 而登录官网​​https://pytorch.org/get-started/locally/​​ ​

参考文章:

​深度学习环境配置​