Can't use cuda.

Hi, I use conda install -c prigoyal pytorch=0.4.0 to install pytorch and run ./demo_test.sh. Everything goes well until line 105 of test.py:

segmentation_module.cuda()

The output is:

 ./demo_test.sh: line 30: 27040 Segmentation fault      (core dumped) python3 -u test.py --model_path $MODEL_PATH --test_img $TEST_IMG --arch_encoder resnet50_dilated8 --arch_decoder ppm_bilinear_deepsup --fc_dim 2048 --result $RESULT_PATH

I guess it’s the problem of pytorch, and test the code:

import torch
x = torch.Tensor(3,3)
x.cuda()

Similar error : Segmentation fault (core dumped).

Any suggestions?

BTW, my other projects of pytorch 0.2.0 and 0.3.0 on this machine work well.

Thanks.

[zhijiew]

Using pytorch 0.4.0 from ostrokach, it can work fine on my machine.

conda install -c ostrokach-forge pytorch

[Tete-Xiao]

@zhiweifang It is because PyTorch’s source codes are still being aggressively changed…

We tried to catch up with the latest source codes, but realized that if we use the newest codes, it doesn’t work on any older version…

We will update and fix warnings when a stable version of PyTorch is released.