今天使用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安装支持