Skip to content

Commit d0be946

Browse files
committed
update 2
1 parent 39c2b79 commit d0be946

File tree

1 file changed

+14
-13
lines changed

1 file changed

+14
-13
lines changed

script/cuda.sh

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -102,17 +102,24 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" = "cu110" ]; then
102102
sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_amd64.deb
103103
sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
104104
sudo apt update -qq
105-
echo "SEARCH 1"
106-
sudo apt search cuda
107-
echo "SEARCH 2"
108-
sudo apt-cache search cuda
109-
echo "END"
110-
sudo apt install cuda
105+
# echo "SEARCH 1"
106+
# sudo apt search cuda
107+
# echo "SEARCH 2"
108+
# sudo apt-cache search cuda
109+
# echo "END"
110+
sudo apt install cuda-repo-ubuntu1804-11-0-local
111111
sudo apt clean
112112
CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
113113
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
114114
PATH=${CUDA_HOME}/bin:${PATH}
115115
nvcc --version
116+
117+
echo "lib=========="
118+
ls $CUDA_HOME/lib
119+
echo "lib64=========="
120+
ls $CUDA_HOME/lib64
121+
echo "include=========="
122+
ls $CUDA_HOME/include
116123
fi
117124

118125
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
@@ -123,13 +130,7 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
123130

124131
# Install CUDA
125132
wget -nv "${CUDA_URL}/${CUDA_FILE}"
126-
127-
if [ "${IDX}" = "cu110" ]; then
128-
PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_11.0 cuobjdump_11.0 nvprune_11.0 nvprof_11.0 cupti_11.0 cublas_11.0 cublas_dev_11.0 cudart_11.0 cufft_11.0 cufft_dev_11.0 curand_11.0 curand_dev_11.0 cusolver_11.0 cusolver_dev_11.0 cusparse_11.0 cusparse_dev_11.0 npp_11.0 npp_dev_11.0 nvrtc_11.0 nvrtc_dev_11.0 nvml_dev_11.0\" -Wait -NoNewWindow"
129-
else
130-
PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} cusolver_dev_${CUDA_SHORT} curand_dev_${CUDA_SHORT}\" -Wait -NoNewWindow"
131-
fi
132-
133+
PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_${CUDA_SHORT} cuobjdump_${CUDA_SHORT} nvprune_${CUDA_SHORT} cupti_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cudart_${CUDA_SHORT} cufft_dev_${CUDA_SHORT} curand_dev_${CUDA_SHORT} cusolver_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} npp_dev_${CUDA_SHORT} nvrtc_dev_${CUDA_SHORT} nvml_dev_${CUDA_SHORT}\" -Wait -NoNewWindow"
133134
CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT}
134135
PATH=${CUDA_HOME}/bin:$PATH
135136
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH

0 commit comments

Comments
 (0)