ImperialCollegeLondon / recode-perceptions

Convolutional Neural Networks exemplar for ReCoDE
https://imperialcollegelondon.github.io/recode-perceptions/
BSD 3-Clause "New" or "Revised" License
0 stars 2 forks source link

Use environment.yml #24

Closed AdrianDAlessandro closed 2 years ago

AdrianDAlessandro commented 2 years ago

Use this instead of the requirements.txt

name: recode
channels:
  - defaults
  - pytorch
  - conda-forge
dependencies:
  - python=3.7
  - torchaudio==0.9.1
  - cudatoolkit=10.2
  - pytorch==1.9.1
  - torchvision==0.10.1
  - wandb
  - tqdm
  - matplotlib
  - pandas

And also change the environment.sh folder to be simply:

module load anaconda3/personal
module load cuda/10.2
conda env create -f environment.yml