The right way to set up the virtual environment

[DKandrew]

The requirements.txt file is broken. Do not use that one to set up your virtual environment.

Do this

# Create a conda environment first
conda create --name pytorch04

# Enter the environment
conda activate pytorch04

# Install pytorch 
conda install pytorch=0.4.1 cuda90 -c pytorch

Replace everything in the requirements.txt file to the following

matplotlib
pandas
jupyterlab
torchvision==0.2.1

then run pip install -r requirements.txt. Now you should be able to run the infer-main.ipynb and train.py

[mebasoglu]

Hello, thank you for sharing this. I also needed to add Pillow==8.4.0 to requirements.txt.