使用CUDA与pytorch？

8 浏览
0 Comments

使用CUDA与pytorch？

有没有一种可靠的方法可以在整个模型上启用CUDA？

我想在GPU上运行训练。我在一些论坛上发现，我需要对想要使用CUDA的所有内容应用.cuda()（我已经对一切可能的内容应用了它，而不会让程序崩溃）。令人惊讶的是，这使得训练速度更慢。

然后，我发现可以使用torch.set_default_tensor_type('torch.cuda.FloatTensor')来使用CUDA。即使两者都启用了，什么都没有改变。发生了什么？

0