RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:663
|2019. 4. 5. 11:58
728x90
반응형
아래와 같이 torch 버전을 다시 설치함으로써 해결
pip3 install -U https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl --user
참고 사이트 : https://discuss.pytorch.org/t/cuda-runtime-error-11/30080/12
728x90
반응형