Linux下 RTX 3090 CUDA和pytorch版本不匹配的问题

conda list

查看pytorch版本,一定要安装到对应的路径下,别一直在base下安装!
在这里插入图片描述在这里插入图片描述
pytorch版本太低,不支持3090的算力,需要重新安装pytorch
官网:https://pytorch.org/get-started/locally/
Linux下 RTX 3090 CUDA和pytorch版本不匹配的问题_第1张图片把命令复制到终端下载

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

Linux下 RTX 3090 CUDA和pytorch版本不匹配的问题_第2张图片

你可能感兴趣的:(pytorch,linux,深度学习)