File tree 4 files changed +9
-11
lines changed
configs/resnet50/vendor_config
docker_images/iluvatar/pytorch
4 files changed +9
-11
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,3 @@ ixrt_tmp_path: iluvatar_tmp/resnet50-fp16.engine
2
2
has_dynamic_axis : false
3
3
repeat : 1
4
4
image_size : 224
5
- exist_onnx_path : onnxs/resnet50_bs256_pytorch_fp16False.onnx
6
- # exist_compiler_path: resnet50-fp16.engine
7
- output_types : {"output":"float32"}
8
- input_types : {"input": "float32"}
Original file line number Diff line number Diff line change @@ -38,9 +38,9 @@ RUN apt-get install -y --fix-missing \
38
38
39
39
40
40
# Configure anaconda
41
- RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py38_4.10.3 -Linux-x86_64.sh && \
42
- bash ./Miniconda3-py38_4.10.3 -Linux-x86_64.sh -b -p /root/miniconda && \
43
- /root/miniconda/bin/conda clean -tipsy && \
41
+ RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0 -Linux-x86_64.sh && \
42
+ bash ./Miniconda3-py310_23.5.2-0 -Linux-x86_64.sh -b -p /root/miniconda && \
43
+ /root/miniconda/bin/conda clean -tip && \
44
44
ln -s /root/miniconda/etc/profile.d/conda.sh /etc/profile.d/conda.sh && \
45
45
echo ". /root/miniconda/etc/profile.d/conda.sh" >> ~/.bashrc && \
46
46
echo "conda activate base" >> ~/.bashrc && \
Original file line number Diff line number Diff line change 2
2
3
3
> 联系邮箱: contact-us@iluvatar.com
4
4
5
- ixrt-0.4 .0+corex.3.2.0-cp38-cp38 -linux_x86_64.whl
5
+ ixrt-0.7 .0+corex.latest.version-cp310-cp310 -linux_x86_64.whl
6
6
7
- torch-1.13 .1+corex.3.2.0-cp38-cp38 -linux_x86_64.whl
7
+ torchvision-0.14 .1+corex.3.2.0.20230914.859-cp310-cp310 -linux_x86_64.whl
8
8
9
- torchvision-0.14.1+corex.3.2.0-cp38-cp38-linux_x86_64.whl
9
+ pycuda-2022.2.2+corex.3.2.0.20230914.859-cp310-cp310-linux_x86_64.whl
10
+
11
+ torch-1.13.1+corex.3.2.0.20230914.859-cp310-cp310-linux_x86_64.whl
Original file line number Diff line number Diff line change 14
14
search_sdk_results=` find ${SDK_DIR} -name " corex*.run" `
15
15
for installer in $search_sdk_results ; do
16
16
echo " Install ${installer} "
17
- sh " ${installer} " -- --silent --driver -- toolkit
17
+ sh " ${installer} " -- --silent --toolkit
18
18
done
19
19
20
20
search_packages_results=` find ${PKG_DIR} -name " *.whl" `
You can’t perform that action at this time.
0 commit comments