相关疑难解决方法(0)

如果我没有指定使用 CPU/GPU,那么我的脚本使用的是哪一个?

在 pytorch 中,如果我没有写任何关于使用 CPU/GPU 的内容,并且我的机器支持 CUDA ( torch.cuda.is_available() == True):

  1. 我的脚本使用的是 CPU 还是 GPU?
  2. 如果是 CPU,我应该怎么做才能让它在 GPU 上运行?我需要重写所有内容吗?
  3. 如果是 GPU,这个脚本会崩溃torch.cuda.is_available() == False吗?
  4. 这对加快训练速度有帮助吗?
  5. 我知道将PyTorch 代码从 CPU 移植到 GPU,但这是旧的。这种情况会在 v0.4 或即将到来的 v1.0 中改变吗?

python pytorch

10
推荐指数
3
解决办法
1万
查看次数

如何在cuda中默认运行pytorch计算

现在,我想使用cuda运行pytorch,然后我使用model.cuda(),并且torch.cuda.LongTensor()用于所有张量..cuda.如果我使用过,我是否必须明确使用张量model.cuda()?有没有办法让所有计算在GPU中运行为默认值?

pytorch

7
推荐指数
1
解决办法
4227
查看次数

标签 统计

pytorch ×2

python ×1