Skip to content

GitLab

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

fix

parent 09b832e1
Show 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}" != "
PATH=${CUDA_HOME}/bin:${PATH}
nvcc --version
echo '----------'
sudo find / -name "*cublas*"
echo '----------'
# Fix cublas on CUDA 10.1:
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
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
......
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