为 Faster-RCNN 设置 CUDA 和 Pytorch

为 Faster-RCNN 设置 CUDA 和 Pytorch

我使用 conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge 设置了 cuda 和 pytorch。但是当我运行 git 代码时,它告诉我:RuntimeError: 未使用 GPU 支持进行编译。

导入 torch 打印(torch.version.cuda) 11.3 torch.cuda.is_available()True

输出 nvidia-smi conda 列表 --show-channel-urls

我的 $CUDA_HOME 未设置,但我不知道应该将其设置为多少。有人能帮我做什么吗?我以前从未使用过 cuda,因此找不到解决问题的方法。

谢谢你!!

答案1

您的屏幕截图显示您已12.2作为您的 CUDA 版本,但您正在使用 11.3 CUDA 版本安装 PyTorch(通过使用标志cudatoolkit=11.3)。检查一下:https://discuss.pytorch.org/t/pytorch-for-cuda-12/169447

相关内容