各种BUG

Error:

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

Solution:

显卡用的最新的RTX2080Ti,pytorch的安装不能直接pip install ,需要从源码编译或者下载wheel安装.

pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

Reference:

  • https://discuss.pytorch.org/t/cuda-runtime-error-11/30080/13
  • https://github.com/pytorch/pytorch/issues/15797#issuecomment-452021037

Error:

The “freeze_support()” line can be omitted if the program is not going to be frozen to produce an executable.

 Solution:

if __name__ == '__main__':
	#your code

 

你可能感兴趣的:(DeepLearning)