-
Installing CUDA 9 requires installation of gcc6. (This first step may
take couple of hours.)
yaourt -S gcc6
-
Remove any existing installations of CUDA
sudo pacman -R cuda pacaur -S chromedriver
-
Download CUDA 9 package from Arch Linux archives
wget https://archive.archlinux.org/packages/c/cuda/cuda-9.0.176-4-x86_64.pkg.tar.xz
-
Install CUDA 9 from local package (Requires Step 1 to be complete)
sudo pacman -U cuda-9.0.176-4-x86_64.pkg.tar.xz
-
Prevent CUDA from upgrading to version 10.
Uncomment the IgnorePKG line to add cuda to make the linesudo vim /etc/pacman.conf
IgnorePKg = cuda
-
Download and install CUDNN 7.0.4-2
wget https://archive.archlinux.org/packages/c/cudnn/cudnn-7.1.4-1-x86_64.pkg.tar.xz sudo pacman -U cudnn-7.1.4-1-x86_64.pkg.tar.xz
-
Install TensorFlow in your virtual environment
pip install tensorflow-gpu
- Rejoice!
Installing TensorFlow for Python on Arch Linux (with CUDA 9 and CUDNN 7)
Published: