RuntimeError: cuda runtime error (78) : a PTX JIT compilation failed at /pytorch/torch/csrc/generic/serialization.cpp:137

jit
1

Hi everyone,
I have an error, the information is as follows

THCudaCheck FAIL file=/pytorch/torch/csrc/generic/serialization.cpp line=137 error=78 : a PTX JIT compilation failed
Traceback (most recent call last):
  File "test_video.py", line 221, in <module>
    main()
  File "test_video.py", line 110, in main
    net.load_state_dict(torch.load(trained_model))
  File "/home/sean/anaconda3/envs/pt40/lib/python3.6/site-packages/torch/serialization.py", line 303, in load
    return _load(f, map_location, pickle_module)
  File "/home/sean/anaconda3/envs/pt40/lib/python3.6/site-packages/torch/serialization.py", line 476, in _load
    deserialized_objects[key]._set_from_file(f, offset, f_is_real_file)
RuntimeError: cuda runtime error (78) : a PTX JIT compilation failed at /pytorch/torch/csrc/generic/serialization.cpp:137

Environment information are:

  1. Pytorch0.40
  2. cuda-8.0
  3. Nvidia Titan X, Nvidia Titan Xp

Could you help me solve it. Thank you in advance.