代码运行报错:THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument
报错原因:显卡用的RTX 2060,CUDA就要装10以上,这个时候pytorch不能直接用pip装,要这样:
pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
不过运行时报错:
ERROR: torch-1.0.0-cp36-cp36m-linux_x86_64.whl is not a supported wheel on th