使用 Visual Studio 代码在 GPU 中运行 PyTorch答案

【问题标题】：Run PyTorch in GPU with visual studio code使用 Visual Studio 代码在 GPU 中运行 PyTorch
【发布时间】：2021-06-07 16:08:32
【问题描述】：

我正在尝试在 Visual Studio 代码中使用 pytorch 执行代码，问题是我必须能够从 CPU 执行此操作。但我的想法是，对于某些深度学习项目使用 gpu，而其他项目不使用。如何从 CPU 切换到 GPU 当我跑步时

import torch
torch.cuda.is_available()

输出是 “假”。

我已经安装了 cuda。我正在使用 Ubuntu 20.04.2 。在 Visual Studio Code 中完成对我来说很重要

【问题讨论】：

【解决方案1】：

有几个问题可能会阻止您使用 GPU。

【讨论】：