尝试在 pyTorch 中使用 Cuda答案

【问题标题】：Trying to use Cuda in pyTorch尝试在 pyTorch 中使用 Cuda
【发布时间】：2020-07-07 16:49:36
【问题描述】：

我一直在研究 PyTorch，想使用 Cuda 张量，但我一直无法让它工作。

当我使用 torch.cuda.is_available() 行时，它返回 false。

但是当我在 anaconda 命令提示符下使用同一行时，它返回 true。是什么赋予了？我需要以某种方式设置设备吗？或者也许让解释器包括我的 GPU？我只想让我的 GPU 被识别为 CUDA 可用并且可以在代码中使用。

我用：

python 3.7

CUDA 10.1

GPU：GTX 1070

【问题讨论】：

【解决方案1】：

我发现了问题，我的解释器出错了，我更改了它以匹配我的 anaconda 提示解释器。像魅力一样工作

【讨论】：