Skip to content

GitLab

Commit 8db0b9f2 authored by rusty1s's avatar rusty1s
Browse files

fix

parent 09b832e1
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
script/cuda.sh
View file @ 8db0b9f2
...@@ -88,12 +88,16 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ] && [ "${IDX}" != " ...@@ -88,12 +88,16 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ] && [ "${IDX}" != "
PATH=${CUDA_HOME}/bin:${PATH} PATH=${CUDA_HOME}/bin:${PATH}
nvcc --version nvcc --version
echo '----------'
sudo find / -name "*cublas*"
echo '----------'
# Fix cublas on CUDA 10.1: # Fix cublas on CUDA 10.1:
if [ -d "${CUDA_HOME}/targets/x86_64-linux/include" ]; then if [ -d "${CUDA_HOME}/targets/x86_64-linux/include" ]; then
cp "${CUDA_HOME}/targets/x86_64-linux/include/*" "${CUDA_HOME}/include/" sudo cp -r "${CUDA_HOME}/targets/x86_64-linux/include/*" "${CUDA_HOME}/include/"
fi fi
if [ -d "${CUDA_HOME}/targets/x86_64-linux/lib" ]; then if [ -d "${CUDA_HOME}/targets/x86_64-linux/lib" ]; then
cp "${CUDA_HOME}/targets/x86_64-linux/lib/*" "${CUDA_HOME}/lib/" sudo cp -r "${CUDA_HOME}/targets/x86_64-linux/lib/*" "${CUDA_HOME}/lib/"
fi fi
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