Spaces:
Running
on
T4
Running
on
T4
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
|