version mismatch between pytorch and cuda in dockfile
Opened this issue · 0 comments
wlt027 commented
Hi,thanks for your work!
I use the dockerfile to test code.The cuda version in docker container is 10.0,but pytorch version is compile with cuda 9.0.
I change the version and it works!
From https://download.pytorch.org/whl/nightly/cu**90**/torch_nightly.html
To https://download.pytorch.org/whl/nightly/cu**100**/torch_nightly.html