blob: eae46bee7aeb5549e2cddf7bb79d6b59ff0070e2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/env sh
if [[ `uname` == 'Linux' ]]; then
CUDA_VERSION=6-5
CUDA_URL=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1204/x86_64/cuda-repo-ubuntu1204_6.5-14_amd64.deb
CUDA_FILE=/tmp/cuda_install.deb
curl $CUDA_URL -o $CUDA_FILE
dpkg -i $CUDA_FILE
rm -f $CUDA_FILE
apt-get -y update
apt-get -y install \
cuda-core-${CUDA_VERSION} \
cuda-cublas-${CUDA_VERSION} \
cuda-cublas-dev-${CUDA_VERSION} \
cuda-cudart-${CUDA_VERSION} \
cuda-cudart-dev-${CUDA_VERSION} \
cuda-curand-${CUDA_VERSION} \
cuda-curand-dev-${CUDA_VERSION} \
cuda-cusparse-${CUDA_VERSION} \
cuda-cusparse-dev-${CUDA_VERSION}
ln -s /usr/local/cuda-6.5 /usr/local/cuda
fi
|