Pytorch installed via pip does not pick up CUDA

[SergeK]

Installing PyTorch into a virtual Python environment

.venv/bin/pip install numpy torch .venv/bin/python -c "import torch; print(torch.cuda.is_available())" False PyTorch from PyPi doesn’t know where to find the cuda driver, you need to tell tell it explicitly e.g. like so: LD_LIBRARY_PATH=/run/opengl-driver/lib python -c "import torch; ..." This is because torch from PyPi is built specifically for FHS operating systems, i.e. systems that drop their libraries in the global /usr/lib and expose a global /etc/ld.so.conf file which tells them ext…