Update switch_cuda.sh
Browse files- switch_cuda.sh +1 -7
switch_cuda.sh
CHANGED
@@ -28,14 +28,8 @@
|
|
28 |
set -e
|
29 |
|
30 |
|
31 |
-
# ensure that the script has been sourced rather than just executed
|
32 |
-
if [[ "${BASH_SOURCE[0]}" = "${0}" ]]; then
|
33 |
-
echo "Please use 'source' to execute switch-cuda.sh!"
|
34 |
-
exit 1
|
35 |
-
fi
|
36 |
-
|
37 |
INSTALL_FOLDER="/usr/local" # the location to look for CUDA installations at
|
38 |
-
TARGET_VERSION
|
39 |
|
40 |
# if no version to switch to has been provided, then just print all available CUDA installations
|
41 |
if [[ -z ${TARGET_VERSION} ]]; then
|
|
|
28 |
set -e
|
29 |
|
30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
INSTALL_FOLDER="/usr/local" # the location to look for CUDA installations at
|
32 |
+
TARGET_VERSION=12.3 # the target CUDA version to switch to (if provided)
|
33 |
|
34 |
# if no version to switch to has been provided, then just print all available CUDA installations
|
35 |
if [[ -z ${TARGET_VERSION} ]]; then
|