search

chldydgh4687 / 00Self-Linux_trial_and_error

Linux & DeepLearning
0 stars 0 forks source link

[SOLVED][pytorch]THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument #41

Open chldydgh4687 opened 4 years ago

chldydgh4687 commented 4 years ago

link : https://github.com/pytorch/pytorch/issues/15797

    I have the same issue. RTX2080, for CUDA 10.0 and pytorch 1.0.0. Anyone has solved the same problem please provide the information of the solution. It would help a lot of people. Thanks!

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” work for me.

IN MY CASE,

“pip3 install -U https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_64.whl"
it works for me.