From 4805dcb70abe3c3a4334de9e5fecd22796db7232 Mon Sep 17 00:00:00 2001 From: "M. Edward (Ed) Borasky" Date: Wed, 15 May 2024 21:38:59 -0700 Subject: [PATCH] torchvision is no longer installed --- emt_source_code/JupyterLab/2_conda_env.sh | 3 +-- emt_source_code/JupyterLab/test-torchvision.py | 7 ------- 2 files changed, 1 insertion(+), 9 deletions(-) delete mode 100644 emt_source_code/JupyterLab/test-torchvision.py diff --git a/emt_source_code/JupyterLab/2_conda_env.sh b/emt_source_code/JupyterLab/2_conda_env.sh index 3b54b62..b61e805 100755 --- a/emt_source_code/JupyterLab/2_conda_env.sh +++ b/emt_source_code/JupyterLab/2_conda_env.sh @@ -23,10 +23,9 @@ conda activate JupyterLab echo "Listing virtual environment to JupyterLab-list.log" conda list > JupyterLab-list.log -echo "Testing PyTorch, torchaudio and torchvision" +echo "Testing PyTorch and torchaudio" python ./test-pytorch-$COMPUTE_MODE.py python ./test-torchaudio.py -python ./test-torchvision.py echo "Testing for R installation" if [ -x /usr/bin/Rscript ] diff --git a/emt_source_code/JupyterLab/test-torchvision.py b/emt_source_code/JupyterLab/test-torchvision.py deleted file mode 100644 index 03890b5..0000000 --- a/emt_source_code/JupyterLab/test-torchvision.py +++ /dev/null @@ -1,7 +0,0 @@ -# https://chrisalbon.com/code/deep_learning/pytorch/basics/check_if_pytorch_is_using_gpu/ - -import torch -print("torch import successful") -import torchvision -print("torchvision import successful") -print("")