Unverified Commit fc4993cd authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

[feature] Use cu92 & ubuntu1604 for torch 1.3.1 (#524)

* [feature] Use cu92 & ubuntu1604 for torch 1.3.1

* [fix]Use ubuntu-16.04 rather than latest for cu92

* [fix]: soft link cuda path

* Debug cu101 bug

* Update usr/include for cublas with cu101

* Check default nvcc version

* rm :

* add cuda root

* Try to locate cublas_v2.h

* Set cublas version

* Force cublas version

* check usr/local/cuda-10.2

* Add tree

* cp cublas files

* cp cublas files to lib64

* do not tree cuda

* fix path cp error

* cp dir

* Keep using CUDA10.1 only

* recover empty line
parent c3d8eb34
...@@ -167,6 +167,8 @@ jobs: ...@@ -167,6 +167,8 @@ jobs:
sudo apt install -y cuda-${CUDA_SHORT/./-} cuda-cufft-dev-${CUDA_SHORT/./-} sudo apt install -y cuda-${CUDA_SHORT/./-} cuda-cufft-dev-${CUDA_SHORT/./-}
sudo apt clean sudo apt clean
export CUDA_HOME=/usr/local/cuda-${CUDA_SHORT} export CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
sudo cp /usr/local/cuda-10.2/include/* /usr/local/cuda/include
sudo cp -r /usr/local/cuda-10.2/lib64/* /usr/local/cuda/lib64/
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/include:${LD_LIBRARY_PATH} export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/include:${LD_LIBRARY_PATH}
export PATH=${CUDA_HOME}/bin:${PATH} export PATH=${CUDA_HOME}/bin:${PATH}
sudo apt-get install -y ninja-build sudo apt-get install -y ninja-build
......
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