Pytorch .to('cuda') 或 .cuda() 不起作用，只是卡住了

Question

我正在尝试做 pytorch 教程。当我尝试将他们的设备设置为 cuda 时，它不起作用并且我的代码运行卡住了。

具体信息，我使用的是conda环境 蟒蛇3.7.3 火炬 1.3.0 cuda 10.2 (NVIDIA RTX2080TI)

>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.cuda.device_count()
1
>>> torch.cuda.current_device()
0
>>> device = torch.device('cuda:0')
>>> device
device(type='cuda', index=0)
>>> aa = torch.randn(5)
>>> aa = tensor([-2.2084, -0.2700, 0.0921, -1.7678, 0.7642])
>>> aa.to(device)


nothing happens...

谁能帮我解决这个问题？

Answer 1

GeForce 30系列需要cuda 11 =<，所以尽量使用cuda 11 =

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

Answer 2

这发生在 Pytorch 1.3.0 版本中（该版本是在本周发布的）。我也面临这个错误。基本上，当我打电话给.to(device) 时，它只是挂起，什么也不做。

如果您想暂时解决此问题，可以降级到 PyTorch 1.2.0。为此，我跑了：

conda install pytorch=1.2.0 torchvision cudatoolkit=10.2 -c pytorch

我本来只想发表评论，但我没有足够的声誉来评论。