pytorch 报错“THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=663 error=11“解决方案

今天使用pysot测试数据,pytorch出现下面报错:
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1535493744281/work/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument

参考了THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp下面的回答

Yiru_Shen:Okay I have solve the problem. You cannot directly install pytorch, instead “pip3 install -U https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl 180” work for me.

通过pip3 install -U https://download.pytorch.org/whl/cu100/torch-1.0.0-cp37-cp37m-linux_x86_64.whl重新安装pytorch问题得到解决(刚开始是通过conda install pytorch==0.4.1安装的pytorch

PS:可以参考这篇文章python 查看pip支持万能方法查看pip安装支持

你可能感兴趣的:(踩过的坑,pytorch,python,pip)