出现PTX JIT compilation failed相关问题
如果cuda的版本和pytorch对应版本没错的情况下,可参考下面的,错误可能来自conda的安装,
可通过下面的进行源码编译,
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl
solved it for me.
参考:
https://github.com/pytorch/pytorch/issues/21004