如何改变forward_hook输出?

回复

pytorchAlso 发起了问题 • 1 人关注 • 0 个回复 • 517 次浏览 • 2017-07-04 22:37 • 来自相关话题

在不同文件夹引入torch,出现错误

回复

pytorchjack 发起了问题 • 1 人关注 • 0 个回复 • 595 次浏览 • 2017-07-03 22:02 • 来自相关话题

pytorch在合理的情况下,支持多类型操作

回复

pytorchjack 发起了问题 • 1 人关注 • 0 个回复 • 352 次浏览 • 2017-07-03 21:55 • 来自相关话题

在PyTorch中建立自己的损失函数

回复

pytorchptorch 发起了问题 • 1 人关注 • 0 个回复 • 2191 次浏览 • 2017-07-01 22:25 • 来自相关话题

Pytorch如何在张量中查找非零元素

回复

pytorch匿名用户 发起了问题 • 1 人关注 • 0 个回复 • 868 次浏览 • 2017-07-01 22:19 • 来自相关话题

PyTorch中如何使用指定的GPU

pytorchwda 发表了文章 • 0 个评论 • 1017 次浏览 • 2017-06-30 23:17 • 来自相关话题

PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。






有如下两种方法来指定需要使用的GPU。
使用CUDA_VISIBLE_DEVICES,类似tensorflow指定GPU的方法。直接终端中设定:CUDA_VISIBLE_DEVICES=1 python my_script.pypython代码中设定:import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"使用函数 set_device,该函数见 pytorch-master\torch\cuda\__init__.py。import torchtorch.cuda.set_device(id)
 
不过官方建议使用CUDA_VISIBLE_DEVICES,不建议使用 set_device 函数。 查看全部
PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。

hero.png


有如下两种方法来指定需要使用的GPU。
  • 使用CUDA_VISIBLE_DEVICES,类似tensorflow指定GPU的方法。
  • 直接终端中设定:
    CUDA_VISIBLE_DEVICES=1 python my_script.py
  • python代码中设定:
    import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"
  • 使用函数 set_device,该函数见 pytorch-master\torch\cuda\__init__.py。
    import torchtorch.cuda.set_device(id)

 
不过官方建议使用CUDA_VISIBLE_DEVICES,不建议使用 set_device 函数。

何时传递一个数字到nn.CrossEntropyLoss()?

回复

pytorch匿名用户 回复了问题 • 1 人关注 • 1 个回复 • 1783 次浏览 • 2017-06-28 22:40 • 来自相关话题

如何在pytorch中微调模型

回复

pytorch匿名用户 发起了问题 • 1 人关注 • 0 个回复 • 733 次浏览 • 2017-06-27 22:06 • 来自相关话题

pytorch中如何加载不同大小的数据集呢?

回复

pytorch匿名用户 回复了问题 • 1 人关注 • 1 个回复 • 739 次浏览 • 2017-06-26 23:17 • 来自相关话题

如何检查模型的参数个数?

回复

pytorchjack 发起了问题 • 1 人关注 • 0 个回复 • 1094 次浏览 • 2017-06-26 23:13 • 来自相关话题