From 118dd08a98820616397b36d7617f7e2d90256d8d Mon Sep 17 00:00:00 2001
From: krishung5 <krish@nvidia.com>
Date: Thu, 14 Nov 2024 16:56:01 -0800
Subject: [PATCH] Try with python 3.12
---
src/model.py | 6 ------
tools/gen_pb_exec_env.sh | 2 +-
2 files changed, 1 insertion(+), 7 deletions(-)
diff --git a/src/model.py b/src/model.py
index 5a251e5..d8ed413 100755
--- a/src/model.py
+++ b/src/model.py
@@ -207,12 +207,6 @@ def initialize(self, args):
"Torch parallelism settings " + for_model + ": " + setting_msg
)
- # Print out python version
- print(
- f"Python version is {sys.version_info.major}.{sys.version_info.minor}",
- flush=True,
- )
-
self._infer_mode = torch.inference_mode(mode=True)
self._infer_mode.__enter__()
diff --git a/tools/gen_pb_exec_env.sh b/tools/gen_pb_exec_env.sh
index d9faf3a..d01a0d9 100755
--- a/tools/gen_pb_exec_env.sh
+++ b/tools/gen_pb_exec_env.sh
@@ -32,7 +32,7 @@ bash Miniconda3-py312_24.9.2-0-Linux-x86_64.sh -p ./miniconda -b
eval "$(./miniconda/bin/conda shell.bash hook)"
# create conda environment
-conda create -n pt python=3.10 -y
+conda create -n pt python=3.12 -y
conda activate pt
conda install -c conda-forge conda-pack -y