Remove CUDA, keep only pytorch

This commit is contained in:
oobabooga 2023-09-22 08:13:11 -07:00
parent 66363a4d70
commit 86648d4085
2 changed files with 2 additions and 1 deletions

View File

@ -0,0 +1 @@
--listen

View File

@ -122,7 +122,7 @@ def install_dependencies():
# Install Pytorch
if gpuchoice == "a":
run_cmd('conda install -y -k cuda ninja git -c nvidia/label/cuda-11.7.0 -c nvidia && python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117', assert_success=True, environment=True)
run_cmd('conda install -y -k ninja git && python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117', assert_success=True, environment=True)
elif gpuchoice == "b" and not is_macos():
if is_linux():
run_cmd('conda install -y -k ninja git && python -m pip install torch==2.0.1+rocm5.4.2 torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2', assert_success=True, environment=True)