Skip to content

Commit 914ea17

Browse files
committed
Install no cuda stuff, cpu only for tf and torch
Setup on Windows and Ubuntu takes ages because we install gpu enabled versions of tensorflow and torch, while very few students have a gpu. Especially knowing that torch is only there for the Transformers day. - Use tensorflow-cpu - Specify wheel location for torch so it downloads a cpu versions - Remove all unnecessary nvidia-cu... packages This reduces the virtualenv size with 4 Gb! ```bash 9.4G ./lewagon 5.4G ./lwlight ```
1 parent 8f9e270 commit 914ea17

File tree

1 file changed

+15
-14
lines changed

1 file changed

+15
-14
lines changed

specs/releases/linux.txt

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -127,18 +127,18 @@ nltk==3.7
127127
notebook==6.4.12
128128
notebook_shim==0.2.0
129129
numpy==1.23.4
130-
nvidia-cublas-cu12==12.1.3.1
131-
nvidia-cuda-cupti-cu12==12.1.105
132-
nvidia-cuda-nvrtc-cu12==12.1.105
133-
nvidia-cuda-runtime-cu12==12.1.105
134-
nvidia-cudnn-cu12==8.9.2.26
135-
nvidia-cufft-cu12==11.0.2.54
136-
nvidia-curand-cu12==10.3.2.106
137-
nvidia-cusolver-cu12==11.4.5.107
138-
nvidia-cusparse-cu12==12.1.0.106
139-
nvidia-nccl-cu12==2.18.1
140-
nvidia-nvjitlink-cu12==12.2.140
141-
nvidia-nvtx-cu12==12.1.105
130+
# nvidia-cublas-cu12==12.1.3.1
131+
# nvidia-cuda-cupti-cu12==12.1.105
132+
# nvidia-cuda-nvrtc-cu12==12.1.105
133+
# nvidia-cuda-runtime-cu12==12.1.105
134+
# nvidia-cudnn-cu12==8.9.2.26
135+
# nvidia-cufft-cu12==11.0.2.54
136+
# nvidia-curand-cu12==10.3.2.106
137+
# nvidia-cusolver-cu12==11.4.5.107
138+
# nvidia-cusparse-cu12==12.1.0.106
139+
# nvidia-nccl-cu12==2.18.1
140+
# nvidia-nvjitlink-cu12==12.2.140
141+
# nvidia-nvtx-cu12==12.1.105
142142
oauthlib==3.2.2
143143
opt-einsum==3.3.0
144144
packaging==21.3
@@ -223,7 +223,7 @@ tenacity==8.1.0
223223
tensorboard==2.10.1
224224
tensorboard-data-server==0.6.1
225225
tensorboard-plugin-wit==1.8.1
226-
tensorflow==2.10.0
226+
tensorflow-cpu==2.10.0
227227
tensorflow-datasets==4.6.0
228228
tensorflow-estimator==2.10.0
229229
tensorflow-io-gcs-filesystem==0.27.0
@@ -239,7 +239,8 @@ toml==0.10.2
239239
tomli==2.0.1
240240
tomlkit==0.11.5
241241
toolz==0.12.0
242-
torch==2.1.0
242+
# torch 2.1.0 cpu only on the next line
243+
https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.1.0%2Bcpu.cxx11.abi-cp310-cp310-linux_x86_64.whl
243244
tornado==6.2
244245
TPOT==0.11.7
245246
tqdm==4.64.1

0 commit comments

Comments
 (0)