Create conda env conda env create -f conda_env_gpu.yaml -n myenv conda activate myenv To run the notebook Replace the checkpoint and data directories by the correct paths. The dataset used is the Task9 Spleen dataset.