我有一个RTX 3060 GPU，但print(torch.cuda.is_available())返回false。我该怎么办？我还在cmd中运行了nvidia-smi命令，结果是：NVIDIA SMI 512.36驱动程序版本：512.36 CUDA版本：11.6那么我应该如何处理这个问题呢？顺便说一句，我正在使用PyCharm。
print(torch.cuda.is_available())
ovfsdjhp1#
你下载了正确版本的PyTorch吗？https://pytorch.org/get-started/locally/并选择您想要的CUDA版本等。据我所知，PyTorch包含自己的cuda版本，不需要本地安装cuda，这也意味着PyTorch不受本地安装cuda的影响。
1条答案
按热度按时间ovfsdjhp1#
你下载了正确版本的PyTorch吗？https://pytorch.org/get-started/locally/并选择您想要的CUDA版本等。
据我所知，PyTorch包含自己的cuda版本，不需要本地安装cuda，这也意味着PyTorch不受本地安装cuda的影响。