Menu
The Tensorflow linux installation instructions say:
Ensure that you create the CUDA_HOME environment variable as described in the NVIDIA documentation.
I cannot find any mention of CUDA_HOME in the NVIDIA instructions for cuDNN v6 or in the NVIDIA CUDA Toolkit install instructions. Does anyone know how this variable should be set on linux?
948
By default, the CUDA SDK Toolkit is installed under /usr/local/cuda/. The nvcc compiler driver is installed in /usr/local/cuda/bin, and the CUDA 64-bit runtime libraries are installed in /usr/local/cuda/lib64. Add /usr/local/cuda/bin to your PATH environment variable.
Environment variable CUDA_HOME , which points to the directory of the installed CUDA toolkit (i.e. /home/user/cuda-10 ) System-wide installation at exactly /usr/local/cuda on Linux platforms. Versioned installation paths (i.e. /usr/local/cuda-10.0 ) are intentionally ignored.
Building applications. The various CUDA Toolkit components are installed in the conda environment at: $CONDA_PREFIX/bin - CUDA executables: nvcc, cuda-memcheck, cuda-gdb, etc.
Run the following command in the terminal:
export CUDA_HOME=/usr/local/cuda-X.X Where you replace X.X by the first two digits of your version number (can be found out e.g. via nvcc --version).
74
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With
