Foi o que eu fiz. Pode haver coisas extras que eu provavelmente não precisei fazer, mas vou incluí-las de qualquer maneira.
Primeiro, obtenha os drivers do repositório ppa. (Eu diria que isso é necessário antes da instalação, a menos que você queira arriscar algum loop de login da morte).
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
Em seguida, instale o driver mais recente. Eu uso o atualizador de GUI na maioria das vezes chamado Software & Updates, na guia Drivers Adicionais. Atualmente, o driver nvidia 396 está disponível.
Obtenha g ++ - 6 e gcc-6. (Requeridos)
sudo apt install g++-6
sudo apt install gcc-6
Você pode tentar usar o nvidia-cuda-toolkit, mas os caminhos para as bibliotecas não me eram familiares. Eu não queria mexer com isso.
(Eu provavelmente pularia este, mas continuaria se outras coisas estivessem dando problemas)
sudo apt install nvidia-cuda-toolkit
Acabei instalando o arquivo de execução 9.1 para o Ubuntu 17.04. Baixe. Marque o arquivo como executável (clique com o botão direito do mouse no arquivo na área de trabalho). Entre no terminal e coloque. (Requeridos)
./cuda_9.1.85_387.26_linux.run --override
Ele o instalará usando os novos compiladores gcc. Ele fará várias perguntas e as responderá imediatamente.
Responda sim à configuração não suportada.
Não para o driver da nvidia.
Sim - ao kit de ferramentas
Eu usei locais de instalação padrão
Após a instalação, certifique-se de configurar seus caminhos. O arquivo de execução também fornecerá um lembrete. A documentação da nvidia também informa o que exportar.
echo 'export PATH=/usr/local/cuda-9.1/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64:$LD_LIBRARY_PATH' >> ~/.bash.rc
source ~/.bashrc
Finalmente, você precisa configurar os simlinks para gcc-6 e g ++ - 6 ou receberá um aviso ao compilar seu próprio código.
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
Reinicie o sistema. (Requeridos)