Container tips


June 14, 2023

Updating Cuda

From docs:

If you want to use a different version of cuda, then you can follow the below steps.

apt update
export DEBIAN_FRONTEND=noninteractive
apt-get install libxml2 -y

Download the required version of Cuda wget

Install Cuda sh --silent --toolkit --toolkitpath=/home/cuda_11.7

You can install a particular version of torch like this pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url