- Run
sudo apt-get update && sudo apt-get upgrade
- Run
./setup_cuda.sh
and follow the prompts:- Say no to 32-bit driver support and DKMS
- Refresh bash
source ~/.bashrc
- Run
./setup_python.sh
- NOTE: you may have to run a 2nd time after doing
~/.bashrc
- NOTE: you may have to run a 2nd time after doing
- Run
source activate cuda
to enter conda environment - (optional) Run
./pip_deps.sh
- Run
./setup_pytorch.sh
You can check CUDA installation by typing nvidia-smi
. If the installation fails, you should run ./purge_cuda.sh
and start again.