Pytorch .to ('cuda') или .cuda () не работает и просто застревает

Question

Я пытаюсь сделать урок по Pytorch. Когда я пытаюсь настроить их устройство как cuda, оно не работает, и мой код работает застрявшим.

Для конкретной информации я использую среду conda из python 3.7.3 pytorch 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

Это произошло с выпуском Pytorch 1.3.0 (релиз был на этой неделе). Я тоже сталкиваюсь с этой ошибкой. По сути, когда я звоню .to(device), он просто зависает и ничего не делает.

Если вы хотите временно это исправить, вы можете перейти на PyTorch 1.2.0. Для этого я набрал:

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

Я бы только прокомментировал, но у меня недостаточно репутации, чтобы сделать это.