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

728x90

[TensorRT] SampleMNIST using RTX 2080 & TensorRT 5.0.2.6 (0)	2019.04.09
onnx.onnx_cpp2py_export.checker.ValidationError: Node (086_upsample) has input size 1 not in range [min=2, max=2] (0)	2019.04.09
ModuleNotFoundError: No module named 'pip._internal' (0)	2019.03.06
AttributeError: module 'tensorflow' has no attribute 'constant' (0)	2019.03.05
Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. (11)	2019.03.05

티스토리툴바