From 1e656bef253d566da620fa2a16c6e7083e2b60ed Mon Sep 17 00:00:00 2001 From: jllllll Date: Wed, 5 Apr 2023 16:52:05 -0500 Subject: [PATCH] Specifically target cuda 11.7 ver. of torch 2.0.0 Move conda-forge channel to global list of channels Hopefully prevents missing or incorrect packages --- install.bat | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/install.bat b/install.bat index 6a8c1f1..6748a84 100644 --- a/install.bat +++ b/install.bat @@ -17,8 +17,8 @@ set /p "gpuchoice=Input> " set gpuchoice=%gpuchoice:~0,1% if /I "%gpuchoice%" == "A" ( - set "PACKAGES_TO_INSTALL=python=3.10.9 torchvision torchaudio pytorch-cuda=11.7 cuda-toolkit conda-forge::ninja conda-forge::git" - set "CHANNEL=-c pytorch -c nvidia/label/cuda-11.7.0 -c nvidia" + set "PACKAGES_TO_INSTALL=python=3.10.9 pytorch[version=2,build=py3.10_cuda11.7*] torchvision torchaudio pytorch-cuda=11.7 cuda-toolkit ninja git" + set "CHANNEL=-c pytorch -c nvidia/label/cuda-11.7.0 -c nvidia -c conda-forge" ) else if /I "%gpuchoice%" == "B" ( set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio cpuonly git" set "CHANNEL=-c conda-forge -c pytorch"