Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-cluster
Commits
d0be9466
Commit
d0be9466
authored
Oct 30, 2020
by
rusty1s
Browse files
update 2
parent
39c2b791
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
14 additions
and
13 deletions
+14
-13
script/cuda.sh
script/cuda.sh
+14
-13
No files found.
script/cuda.sh
View file @
d0be9466
...
@@ -102,17 +102,24 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" = "cu110" ]; then
...
@@ -102,17 +102,24 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" = "cu110" ]; then
sudo
dpkg
-i
cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_amd64.deb
sudo
dpkg
-i
cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_amd64.deb
sudo
apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
sudo
apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
sudo
apt update
-qq
sudo
apt update
-qq
echo
"SEARCH 1"
#
echo "SEARCH 1"
sudo
apt search cuda
#
sudo apt search cuda
echo
"SEARCH 2"
#
echo "SEARCH 2"
sudo
apt-cache search cuda
#
sudo apt-cache search cuda
echo
"END"
#
echo "END"
sudo
apt
install
cuda
sudo
apt
install
cuda
-repo-ubuntu1804-11-0-local
sudo
apt clean
sudo
apt clean
CUDA_HOME
=
/usr/local/cuda-
${
CUDA_SHORT
}
CUDA_HOME
=
/usr/local/cuda-
${
CUDA_SHORT
}
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
LD_LIBRARY_PATH
=
${
CUDA_HOME
}
/lib64:
${
LD_LIBRARY_PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
PATH
=
${
CUDA_HOME
}
/bin:
${
PATH
}
nvcc
--version
nvcc
--version
echo
"lib=========="
ls
$CUDA_HOME
/lib
echo
"lib64=========="
ls
$CUDA_HOME
/lib64
echo
"include=========="
ls
$CUDA_HOME
/include
fi
fi
if
[
"
${
TRAVIS_OS_NAME
}
"
=
"windows"
]
&&
[
"
${
IDX
}
"
!=
"cpu"
]
;
then
if
[
"
${
TRAVIS_OS_NAME
}
"
=
"windows"
]
&&
[
"
${
IDX
}
"
!=
"cpu"
]
;
then
...
@@ -123,13 +130,7 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
...
@@ -123,13 +130,7 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
# Install CUDA
# Install CUDA
wget
-nv
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
wget
-nv
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
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"
if
[
"
${
IDX
}
"
=
"cu110"
]
;
then
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"
else
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"
fi
CUDA_HOME
=
/c/Program
\
Files/NVIDIA
\
GPU
\
Computing
\
Toolkit/CUDA/v
${
CUDA_SHORT
}
CUDA_HOME
=
/c/Program
\
Files/NVIDIA
\
GPU
\
Computing
\
Toolkit/CUDA/v
${
CUDA_SHORT
}
PATH
=
${
CUDA_HOME
}
/bin:
$PATH
PATH
=
${
CUDA_HOME
}
/bin:
$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
PATH
=
/c/Program
\
Files
\ \(
x86
\)
/Microsoft
\
Visual
\
Studio/2017/BuildTools/MSBuild/15.0/Bin:
$PATH
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment