Files
CSE5100H2/hw2/README.md
Zheyuan Wu 250f763f1f done?
2025-10-12 00:55:07 -05:00

1.1 KiB

Installation

Since we are using PyTorch for hw2, we recommend using conda to manage the environment. Please refer to the miniconda homepage for a compact conda installation.

You have two options for creating the environment of hw2

  • For mac users or a cpu-only installation, please remove the pytorch-cuda term in either ways.

  • To create a new conda environment, simply run conda env create -f environment.yml

  • If you want to install the package within the environment you created with hw1, please following the below steps:

    conda activate <hw1-env-name>
    # we are using PyTorch 2.0!
    # remove the pytorch-cuda=11.7 term if you are a mac user to want a cpu-only installation
    conda install pytorch==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia
    pip install gymnasium[classic_control]==0.27.1
    pip install matplotlib==3.7.1
    # for hyperparameter management
    pip install hydra-core==1.3.2
    # for video recording
    pip install moviepy==1.0.3
    

That's it! If you encounter any trouble creating the environment, please let us know :-)