THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument

原因是显卡用的RTX 2080Ti,CUDA就要装10以上,这个时候,请看发生了变化:

https://pytorch.org/resources

页面最下,通过选择可以看到:

pip install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp27-cp27m-linux_x86_64.whl

猜你喜欢

转载自blog.csdn.net/quantum7/article/details/88828080
今日推荐