Skip to content

GitLab

Commit a9f38640 authored by rusty1s's avatar rusty1s
Browse files

prop envs

parent 8bca192e
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
.travis.yml
View file @ a9f38640
...@@ -18,7 +18,7 @@ env: ...@@ -18,7 +18,7 @@ env:
- FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=${TORCH_VERSION} - FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=${TORCH_VERSION}
before_install: before_install:
- bash install.sh - source install.sh
install: install:
- pip install numpy - pip install numpy
......
install.sh
View file @ a9f38640
#!/bin/bash #!/bin/bash
if [ "$TRAVIS_OS_NAME" = "linux" ] && [ -z "${CUDA}" ]; then if [ "$TRAVIS_OS_NAME" = "linux" ] && [ "${FORCE_CUDA}" != "1" ]; then
export INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
wget "http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}" wget "http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}"
sudo dpkg -i "${INSTALLER}" sudo dpkg -i "${INSTALLER}"
wget "https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub" wget "https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub"
...@@ -9,8 +9,8 @@ if [ "$TRAVIS_OS_NAME" = "linux" ] && [ -z "${CUDA}" ]; then ...@@ -9,8 +9,8 @@ if [ "$TRAVIS_OS_NAME" = "linux" ] && [ -z "${CUDA}" ]; then
sudo apt update -qq sudo apt update -qq
sudo apt install -y "cuda-core-${CUDA_SHORT/./-}" "cuda-cudart-dev-${CUDA_SHORT/./-}" "${CUBLAS}" "cuda-cusparse-dev-${CUDA_SHORT/./-}" sudo apt install -y "cuda-core-${CUDA_SHORT/./-}" "cuda-cudart-dev-${CUDA_SHORT/./-}" "${CUBLAS}" "cuda-cusparse-dev-${CUDA_SHORT/./-}"
sudo apt clean sudo apt clean
export CUDA_HOME=/usr/local/cuda-${CUDA_SHORT} CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
export PATH=${CUDA_HOME}/bin:${PATH} PATH=${CUDA_HOME}/bin:${PATH}
nvcc --version nvcc --version
fi fi
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment