Skip to content

GitLab

"torchvision/vscode:/vscode.git/clone" did not exist on "385a44f84859477ffa2d574fa6fbd4930dec9bf6"
Unverified Commit c65edb15 authored by Bruce MacDonald's avatar Bruce MacDonald Committed by GitHub
Browse files

fix linux installer warning logs (#588)

parent 1605af32
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
scripts/install.sh
View file @ c65edb15
...@@ -6,6 +6,7 @@ set -eu ...@@ -6,6 +6,7 @@ set -eu
status() { echo ">>> $*" >&2; } status() { echo ">>> $*" >&2; }
error() { echo "ERROR $*"; exit 1; } error() { echo "ERROR $*"; exit 1; }
warning() { echo "WARNING: $*"; }
TEMP_DIR=$(mktemp -d) TEMP_DIR=$(mktemp -d)
cleanup() { rm -rf $TEMP_DIR; } cleanup() { rm -rf $TEMP_DIR; }
...@@ -95,7 +96,7 @@ fi ...@@ -95,7 +96,7 @@ fi
check_gpu() { check_gpu() {
case $1 in case $1 in
lspci) command -v lspci >/dev/null && lspci -d '10de:' | grep -q 'NVIDIA' || return 1 ;; lspci) command -v lspci >/dev/null && lspci -d '10de:' | grep -q 'NVIDIA' || return 1 ;;
lshw) command -v lshw >/dev/null && lshw -c display -numeric | grep -q 'vendor: .* \[10DE\]' || return 1 ;; lshw) command -v lshw >/dev/null && $SUDO lshw -c display -numeric | grep -q 'vendor: .* \[10DE\]' || return 1 ;;
nvidia-smi) command -v nvidia-smi >/dev/null || return 1 ;; nvidia-smi) command -v nvidia-smi >/dev/null || return 1 ;;
esac esac
} }
......
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