如何在 PyTorch 中为 Nvidia GeForce RTX 3050 Ti 启用 CUDA？

Question

我需要帮助，如果您能帮助我，我将不胜感激。我想在我的显卡（Nvidia GeForce RTX 3050 Ti）上运行 PyTorch 库，该库在 PyCharm 的虚拟环境中运行。但是，它在 CPU 上运行，每当我使用命令 `import torch` 和 `print("cuda is available:", torch.cuda.is_available())` 时，它总是返回 False。

在我的系统上，安装了 CUDA 版本 12.6。我还安装了 CUDA 版本 12.4 的 PyTorch，因为它是 PyTorch 网站上提供的最新版本。请帮助我考虑到我的显卡类型应该安装什么。

Answer 1

我假设您正在谈论在终端中运行 nvidia-smi 命令时提到的 CUDA 版本 12.6。

这显示了您的 GPU 支持的 CUDA 版本，并不意味着您的系统上安装了 CUDA 12.6。

您需要根据操作系统运行以下命令：

在 Windows 系统上：

pip3 install torch --index-url https://download.pytorch.org/whl/cu124

在 Linux 系统上：

pip3 install torch

注意：在运行上述 pip 命令之前，还要确保您的虚拟环境在 pycharm 终端中处于活动状态。