sentence_embedding_pipeline/docker/install_cpu.sh

#!/bin/bash

torch_version=${1:-2.4.0}
torchvision_version=${2:-0.19.0}
torchaudio_version=${3:-2.4.0}

pip uninstall -y torch torchvision torchaudio

pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version --index-url https://download.pytorch.org/whl/cpu
Refactor Dockerfile (#1036) 2024-10-21 18:26:23 +08:00			`#!/bin/bash`

			`torch_version=${1:-2.4.0}`
			`torchvision_version=${2:-0.19.0}`
			`torchaudio_version=${3:-2.4.0}`

Refine docker file (#1041) 2024-10-22 11:13:17 +08:00			`pip uninstall -y torch torchvision torchaudio`

update docker (#1073) 2024-11-08 10:56:39 +08:00			`pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version --index-url https://download.pytorch.org/whl/cpu`