Skip to content

GitLab

Commit 10592cb5 authored by rusty1s's avatar rusty1s
Browse files

pytorch with cuda

parent 06323758
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
.travis.yml
View file @ 10592cb5
...@@ -20,16 +20,18 @@ before_install: ...@@ -20,16 +20,18 @@ before_install:
- 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
- sudo apt-key add 7fa2af80.pub - sudo apt-key add 7fa2af80.pub
- sudo apt update -qq - sudo apt update -qq
- sudo apt install -y cuda-${CUDA_SHORT/./-} - sudo apt install -y cuda-core-${CUDA_SHORT/./-} cuda-cudart-dev-${CUDA_SHORT/./-} cuda-cufft-dev-${CUDA_SHORT/./-}
- sudo apt clean
- CUDA_HOME=/usr/local/cuda-${CUDA_SHORT} - CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
- LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} - LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
- PATH=${CUDA_HOME}/bin:${PATH} - PATH=${CUDA_HOME}/bin:${PATH}
install: install:
- pip install numpy - pip install numpy
- pip install torch - pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
script: script:
- nvcc --version - nvcc --version
- python -c "import torch; print(torch.__version__)" - python -c "import torch; print(torch.__version__)"
- python setup.py install - python setup.py install
- python setup.py test
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