Create a virtual environment, activate it and then run:
conda create -n rubicon_ds_take_home python=3.11
conda activate rubicon_ds_take_home
conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia
pip install -r requirements.txt
Create a virtual environment, activate it and then run:
conda create -n rubicon_ds_take_home python=3.11
conda activate rubicon_ds_take_home
conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia
pip install -r requirements.txt