-
Notifications
You must be signed in to change notification settings - Fork 1
Old cuda hacks
Andy Casey edited this page Dec 31, 2020
·
1 revision
If you get issues like this:
RuntimeError: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.
Then do this:
pip install torch==1.5.1 torchvision==0.6.1
(You want torch
> 1.5 as there were big changes in how models were saved)
And add the following to your ~/.bash_profile
:
export CUDA_VISIBLE_DEVICES=""
Trying:
This seems to have worked:
pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
+/- with CUDA_VISIBLE_DEVICES