| 1234567891011121314151617181920212223 |
- #!/bin/bash
- # Replace '.' with '-' ex: 11.8 -> 11-8
- cuda_version=$(echo $1 | tr "." "-")
- # Removes '-' and '.' ex: ubuntu-20.04 -> ubuntu2004
- OS=$(echo $2 | tr -d ".\-")
- # Installs CUDA
- wget -nv https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-keyring_1.1-1_all.deb
- sudo dpkg -i cuda-keyring_1.1-1_all.deb
- rm cuda-keyring_1.1-1_all.deb
- sudo apt -qq update
- sudo apt -y install cuda-${cuda_version} cuda-nvcc-${cuda_version} cuda-libraries-dev-${cuda_version}
- sudo apt clean
- # Test nvcc
- PATH=/usr/local/cuda-$1/bin:${PATH}
- nvcc --version
- # Log gcc, g++, c++ versions
- gcc --version
- g++ --version
- c++ --version
|
