You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
934 B
20 lines
934 B
#!/usr/bin/env bash
|
|
#
|
|
# Install the core CUDA_VER toolkit for a ubuntu-trusty (14.04) system. Requires the
|
|
# CUDA_VER environment variable to be set to the required version.
|
|
#
|
|
# Since this script updates environment variables, to execute correctly you must
|
|
# 'source' this script, rather than executing it in a sub-process.
|
|
#
|
|
# Taken from https://github.com/tmcdonell/travis-scripts.
|
|
|
|
export CUDA_VER=8.0.61-1
|
|
travis_retry wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_${CUDA_VER}_amd64.deb
|
|
travis_retry sudo dpkg -i cuda-repo-ubuntu1404_${CUDA_VER}_amd64.deb
|
|
travis_retry sudo apt-get update -qq
|
|
export CUDA_APT=${CUDA_VER:0:3}
|
|
export CUDA_APT=${CUDA_APT/./-}
|
|
travis_retry sudo apt-get install -qy cuda-drivers cuda-core-${CUDA_APT} cuda-cudart-dev-${CUDA_APT}
|
|
travis_retry sudo apt-get clean
|
|
export CUDA_HOME=/usr/local/cuda-${CUDA_VER:0:3}
|
|
export PATH=${CUDA_HOME}/bin:${PATH}
|
|
|