mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[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
pull/527/head
Wenwei Zhang 2020-09-01 16:15:03 +08:00 committed by GitHub
parent c3d8eb34ff
commit fc4993cdec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/build.yml vendored
View File

@ -167,6 +167,8 @@ jobs:
sudo apt install -y cuda-${CUDA_SHORT/./-} cuda-cufft-dev-${CUDA_SHORT/./-}
sudo apt clean
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 PATH=${CUDA_HOME}/bin:${PATH}
sudo apt-get install -y ninja-build