I installed pytorch nightly for cuda10 using pip3 by running:
pip3 install torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html 4
Now whenever I try to import torchvision I get:
from torchvision import _C
ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory
Cuda 10 is installed on my system and is symlinked to /usr/local/cuda/lib64
Here is the output of nvidia-smi

image.png778×358 36.3 KB

What went wrong? Why is torchvisions looking for cuda 9.0 files?