Iluvatar Ixrt environment (#265)

* Ixrt environment * add touch config --------- Co-authored-by: 杨智超 <[email protected]>
FlagOpen · Sep 22, 2023 · 3d4c437 · 3d4c437
1 parent a8599e7
commit 3d4c437
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 11 deletions.
diff --git a/inference/configs/resnet50/vendor_config/iluvatar_configurations.yaml b/inference/configs/resnet50/vendor_config/iluvatar_configurations.yaml
@@ -2,7 +2,3 @@ ixrt_tmp_path: iluvatar_tmp/resnet50-fp16.engine
 has_dynamic_axis: false
 repeat: 1
 image_size: 224
-exist_onnx_path: onnxs/resnet50_bs256_pytorch_fp16False.onnx
-# exist_compiler_path: resnet50-fp16.engine
-output_types: {"output":"float32"}
-input_types: {"input": "float32"}
diff --git a/inference/docker_images/iluvatar/pytorch/Dockerfile b/inference/docker_images/iluvatar/pytorch/Dockerfile
@@ -38,9 +38,9 @@ RUN apt-get install -y --fix-missing \
 
 
 # Configure anaconda
-RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py38_4.10.3-Linux-x86_64.sh && \
-    bash ./Miniconda3-py38_4.10.3-Linux-x86_64.sh -b -p /root/miniconda && \
-    /root/miniconda/bin/conda clean -tipsy && \
+RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh && \
+    bash ./Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -b -p /root/miniconda && \
+    /root/miniconda/bin/conda clean -tip && \
     ln -s /root/miniconda/etc/profile.d/conda.sh /etc/profile.d/conda.sh && \
     echo ". /root/miniconda/etc/profile.d/conda.sh" >> ~/.bashrc && \
     echo "conda activate base" >> ~/.bashrc && \

diff --git a/inference/docker_images/iluvatar/pytorch/packages/README.md b/inference/docker_images/iluvatar/pytorch/packages/README.md
@@ -2,8 +2,10 @@
 
 >联系邮箱: [email protected]
 
-ixrt-0.4.0+corex.3.2.0-cp38-cp38-linux_x86_64.whl
+ixrt-0.7.0+corex.latest.version-cp310-cp310-linux_x86_64.whl
 
-torch-1.13.1+corex.3.2.0-cp38-cp38-linux_x86_64.whl
+torchvision-0.14.1+corex.3.2.0.20230914.859-cp310-cp310-linux_x86_64.whl
 
-torchvision-0.14.1+corex.3.2.0-cp38-cp38-linux_x86_64.whl
+pycuda-2022.2.2+corex.3.2.0.20230914.859-cp310-cp310-linux_x86_64.whl
+
+torch-1.13.1+corex.3.2.0.20230914.859-cp310-cp310-linux_x86_64.whl
diff --git a/inference/docker_images/iluvatar/pytorch/pytorch_install.sh b/inference/docker_images/iluvatar/pytorch/pytorch_install.sh
@@ -14,7 +14,7 @@ done
 search_sdk_results=`find ${SDK_DIR} -name "corex*.run"`
 for installer in $search_sdk_results; do
     echo "Install ${installer}"
-    sh "${installer}" -- --silent --driver --toolkit
+    sh "${installer}" -- --silent --toolkit
 done
 
 search_packages_results=`find ${PKG_DIR} -name "*.whl"`