I am trying to change my environment path variables so Pytorch can access CUDA. According to nvidia-smi, the driver can support CUDA 10.0. CUDA 10.0 is already installed on the server. I modified my bash_profile according to these instructions.
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64\{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
nvcc shows the CUDA version. However, Pytorch still does not detect CUDA.
Is there some solution? I am a non-root user.