python出现THCudaCheck FAIL file=/pytorch/aten/src/THC/ThCGeneral.cpp line=405 error=11 : invalid argument情况时，

问题：cuda版本对应的pytorch版本不符合，

解决方案：安装对应cuda版本的pytorch（我用的是cuda11.0，如果使用其他版本请依次更换）

进入conda环境后，用pip安装pytorch

pip install https://download.pytorch.org/whl/cu110/torch-1.7.0%2Bcu110-cp37-cp37m-linux_x86_64.whl

pip install torchvision

问题解决！