Installation Of Tensorflow Pytorch Gpu
2018-09-05
Environment
Arch Linux
GPU driver
- Download driver from GPU vendor such as nvidia.
- Switch to tty mode.
- Run the install script by
sudo sh script.run
. - Run
nvidia-modprobe
anddmesg
to verify installation.
note:need to disable nouveau driver by the script if installed.
Install TensorFlow-cuda
- Run
sudo pacman -Sy python-tensorflow-cuda
.(or by pip or by conda) - Clone tensorflow models from https://github.com/tensorflow/models.git.
- Run a model to verify installation.
Install Pytorch-cuda
- Run
sudo pacman -Sy python-pytorch-cuda
.(or by pip or conda) - Clone tensorflow models from https://github.com/pytorch/examples.git.
- Run a model to verify installation.
note:install in the way just need to pay attention to conflict between driver and pytorch/tensorflow.