A100-PCIE-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation

使用以下命令进行安装,成功安装运行:

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(CUDA/GPU,pytorch,深度学习,python)