Skip to content
Snippets Groups Projects
Commit aa3afd7e authored by Xuan Gu's avatar Xuan Gu
Browse files

Update 2 files

- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
parent 0a3e326b
No related branches found
No related tags found
No related merge requests found
......@@ -14,7 +14,7 @@ apptainer build $CONTAINER_DIR $DEF_DIR
MODULE_NAME=nnunet_for_pytorch
MODULE_VERSION=latest
CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
apptainer build $CONTAINER_DIR $DEF_DIR
......
......@@ -14,7 +14,7 @@ VERSION=latest
mkdir /workspace/nnunet_pyt
cd /workspace/nnunet_pyt
git clone https://github.com/NVIDIA/DeepLearningExamples.git
mv DeepLearningExamples/PyTorch/Segmentation/nnUNet ./
mv DeepLearningExamples/PyTorch/Segmentation/nnUNet/* ./
rm -rf DeepLearningExamples
pip install --disable-pip-version-check -r requirements.txt
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment