install-pytorch-1.11.0-cuda-11.3.sh

#!/bin/bash
# Install PyTorch into a conda environment

# Retrieve the script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Configuration
export CFG_CUDA_VERSION=11.3
export CFG_PYTORCH_TAG=v1.11.0
export CFG_TORCHVISION_TAG=v0.12.0
export CFG_TORCHAUDIO_TAG=e92a17c35fdff6b0622b0791b43e665c5d05c4b4  # One commit after v0.11.0
export CFG_TORCHAUDIO_VERSION=v0.11.0+e92a17c
export CFG_TORCHTEXT_TAG=v0.12.0
export CFG_OPENCV_TAG=4.5.5
export CFG_CONDA_PYTHON=3.9

# Run the main installation script
"$SCRIPT_DIR/install-pytorch.sh"
# EOF