Pytorch 0.4.0 issue

[blueclowd]

Sharing with this issue on Pytorch 0.4.0

Traceback (most recent call last):
File "train_textsnake.py", line 241, in
main()
File "train_textsnake.py", line 226, in main
train(model, train_loader, criterion, scheduler, optimizer, epoch, logger)
File "train_textsnake.py", line 66, in train
for i, (img, train_mask, tr_mask, tcl_mask, radius_map, sin_map, cos_map, meta) in enumerate(train_loader):
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 451, in iter
return _DataLoaderIter(self)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 247, in init
self._put_indices()
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 295, in _put_indices
indices = next(self.sample_iter, None)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/sampler.py", line 138, in iter
for idx in self.sampler:
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/sampler.py", line 51, in iter
return iter(torch.randperm(len(self.data_source)).tolist())
RuntimeError: randperm is only implemented for CPU

=======================================================
In my case, I need to install other versions of PyTorch to solve this issue.

[15926273249]

请问这个问题解决了吗

[blueclowd]

Hi, @15926273249

我沒有細究這個issue的原因，不過換了pytorch 的版本就沒有這個問題了。以下是我的安裝指令，給您參考：
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html