Python，Pytorch，在使用 cuda 时遇到问题答案

【问题标题】：Python, Pytorch, having trouble using cudaPython，Pytorch，在使用 cuda 时遇到问题
【发布时间】：2020-07-15 09:44:34
【问题描述】：

操作系统：Windows 10

Python 3.7.4（康达）

GPU：GTX 980

所以我安装了 CUDA 工具包 v.10.1，获得了匹配的 cuDNN 文件，安装了启用 cuda 10.1 的 pytorch 版本，此外我还更新了我的 gpu 驱动程序。但如果我现在尝试检查我的 gpu 是否与torch.cuda.is_available() 一起可用，它仍然会得到False。有什么想法吗？？

【问题讨论】：

您是否传递了任何强制不使用 GPU 的环境变量？
@TommasoBendinelli 我不这么认为。这是整个代码import torch print(torch.cuda.is_available()) 但我不确定我是否理解你的问题。
python（在实际运行代码之前）是否与其他额外的全局环境变量一起调用？我希望现在更清楚了
@TommasoBendinelli 好吧，如果这算作全局环境，我正在使用 Anaconda，但我不记得在那里做任何额外的设置。

标签： python pytorch

【解决方案1】：

您需要确保您的 python 在 GPU 而不是 CPU 上运行。

您可以通过从右键单击选项中选择在 GPU 上启动 IDE 来执行此操作。只需右键单击您的 IDE 并选择 Run with graphics processer。如下图所示。

或者，您可以转到Nvidia Control Panel > Manage 3D Settings > Program Settings 并将 Python 的默认图形处理器和您的 IDE 自定义为 NVIDIA GPU。控制面板如下所示。

【讨论】：

!Error_msg。这是我尝试使用我的 gpu 启动 VS 代码时遇到的错误。另外关于替代方案，它没有将 VS 代码显示为我的程序，我不知道为什么。我也试过重新安装它。
从 https://developer.nvidia.com/nsight-graphics 重新安装 Nvidia NSight Graphics。也可以尝试使用 VS Code 以外的其他 IDE。
您可以使用Add按钮在Nvidia程序设置中添加VS Code并从您的程序文件中找到它的可执行文件。
我刚刚做了这两件事，它仍然返回一个False（在 VS Code 和 Jupyter Notebook 中都试过）。