Torch，如何检查变量是否为CUDA？

文章/答案/技术大牛

发布

1回答

lua、torch

我正在寻找一个像type()这样的函数来识别哪个变量是CudaTensor还是Normal。require('cutorch') x = x:cuda() elseend

浏览 2提问于2016-08-30得票数 2

回答已采纳

13回答

如何检查PyTorch是否正在使用GPU？

python、memory-management、gpu、nvidia、pytorch

如何检查PyTorch是否正在使用GPU？nvidia-smi命令可以检测GPU活动，但我想直接从Python中检查它。

浏览 19提问于2018-01-08得票数 429

回答已采纳

2回答

PyTorch期望CPU得到CUDA张量

python、pytorch、generative-adversarial-network、dcgan

另外，请检查gen_input是否正确。generator.parameters(), lr = lr, betas = (beta1, 0.999)) if torch.cuda.is_available(): discriminator = discriminator.cuda() criter

浏览 3提问于2020-09-30得票数 1

回答已采纳

1回答

google colab的问题-RuntimeError:预期所有张量都在同一设备上，但至少发现了两个设备，cuda:0和cpu

python、pytorch、runtime-error

当我运行下面附加的代码时，会出现这样的错误：mel_gt= get_mel(audio) mel_shift = torch.cat((torch.zeros_like-1) mel_est, attention =

浏览 1提问于2022-01-17得票数 2

1回答

在PyTorch中，当GPU张量被分配给一个新值时，GPU内存是否被释放？

python、pytorch

当PyTorch中的Cuda变量被分配一个新值时，它再次成为一个CPU变量(如下面的代码所示)。在这种情况下，以前GPU上的变量所持有的内存是否被自动释放？import torch t1 = t1.cuda() t1 = torch

浏览 7提问于2020-08-23得票数 4

回答已采纳

2回答

无法导入torch (ImportError: libcudart.so.10.0)

linux、pytorch、nvidia、torch、nvidia-jetson-nano

我正在尝试运行一个python文件，该文件导入了一个名为torch的包。我按照NVIDIA 的说明将其与torchvision一起安装。/jiayi/.local/lib/python3.6/site-packages/torch/__init__.py", line 81, in <module> from torch.它能够检测到安装了torch包，但似乎存在内部问题。Requirement already satisf

浏览 203提问于2020-05-22得票数 0

2回答

如何检查张量是否在库达上，还是将其发送给焦火炬中的库达？

python、pytorch、gpu、tensor

我有张量如何检查它是否在gpu上，并发送到gpu并返回？

浏览 6提问于2020-12-20得票数 15

回答已采纳

1回答

关于torch.nn.DataParallel的几个问题

python、pytorch

由于它们使用多个GPU，所以在代码中有一个命令torch.nn.DataParallel(model, device_ids= args.gpus).cuda()。

浏览 1提问于2018-10-05得票数 4

1回答

如果GPU不存在，请使用CPU

python、tensorflow

self.device = torch.device('cuda‘如果torch.cuda.is_available()其他'cpu') 用于检查是否存在gpu的代码，如果没有，则使用Pytorch来使用如何使用tensorflow实现这一点？

浏览 15提问于2022-04-12得票数 0

5回答

如果我没有指定使用CPU/GPU，我的脚本使用哪一个？

python、pytorch

如果我没有写任何关于使用CPU/GPU的文章，并且我的机器支持CUDA (torch.cuda.is_available() == True)：我知道，但这是旧的。

浏览 2提问于2018-05-23得票数 12

回答已采纳

0回答

在pytorch中自动设置cuda变量

python、pytorch

我正在尝试自动设置变量，以便在标志use_gpu被激活时使用cuda。通常我会这样做：from torch import autogradimport torch from torch import autog

浏览 6提问于2017-12-05得票数 2

回答已采纳

1回答

如何修复"AssertionError: CUDA不可用，请求的设备0无效“

python、deep-learning、pytorch

", line 47, in select_device>>import torch >>print(torch.cuda.is_available()) 我得到了False，它解释了这个

浏览 2818提问于2021-07-28得票数 0

1回答

如何将PyTorch版本和CUDA版本保存到环境变量

docker、dockerfile、environment-variables、torch

//data.pyg.org/whl/torch-${TORCH}+${CUDA}.html因此，我需要在构建时将这些版本的捕获到环境变量TORCH和CUDA中。我知道

浏览 9提问于2022-08-04得票数 0

回答已采纳

1回答

检查预期的对象是后端CUDA还是CPU？

python-3.x、pytorch

我试图在CPU和CUDA上运行一个代码。代码构思： device = torch.device("cuda" if torch.cuda.is_available() else "cpu") i

浏览 1提问于2019-04-23得票数 0

1回答

使用使用gpu的stanfordnlp对python代码进行容器化

python、docker、stanford-nlp

附注：我已经在中尝试了使用部分，并检查了我的服务器是否有3个gpus。

浏览 23提问于2020-03-17得票数 0

回答已采纳

1回答

当未设置-- GPU时，防止在SLURM中使用GPU

pytorch、slurm

当用户通过--gpus=2请求GPU时，SLURM分配给用户的GPU设置为CUDA_VISIBLE_DEVICES环境变量。$ srun --gpus=2 bash -c 'echo $CUDA_VISIBLE_DEVICES'我们有一个小团队，可以相信我们的用户不会滥用系统(他们可以很容易地覆盖环境变量)，所以这很好用设置为-1，它将失败。/conda-bld/pytorch_1556653099582/work/a

浏览 89提问于2019-08-23得票数 5

回答已采纳

1回答

如何在不初始化CUDA的情况下检查火炬gpu的兼容性？

python、tensorflow、pytorch、gpu

在我的例子中，崩溃有以下错误： return torch.尽管如此，这种情况仍然发生： assert torch.

浏览 1提问于2021-08-04得票数 2

回答已采纳

1回答

Torch.cuda.empty_cache()非常慢的性能

pytorch、gpu、inference

active - the bottleneck after the first batch moves here, rather than below # torch.cuda.empty_cacheHIT HAPPENS HERE - after the first batch # obviously tensor.to(device) uses torch.cuda.empty_cachet in bat

浏览 2提问于2021-02-22得票数 0

回答已采纳

1回答

如果GPU没有“热身”的话，Pytorch内存不足吗？

python、pytorch、nvidia

我有一个简单的脚本如下：LOSS_WEIGHTS = [1,2,3]LOSS_WEIGHTS= LOSS_WEIGHTS.to(0) 如果我启动脚本时，计算机是空闲的，我经常得到“CUDA错误:内存不足”，但GPU是完全空的。我应该在启动脚本之前初始化我的cuda设备吗？pytors1.2.0(尝试了几个版本) cuda 10.1 (也尝试了code 9)python 3.7Nvidia驱动器

浏览 3提问于2019-11-13得票数 0

7回答