Skip to content

Commit f1cb4b8

Browse files
committed
Added PyToch 1.10 changes
1 parent 9aa4719 commit f1cb4b8

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

Jetson/Dockerfile

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -194,10 +194,14 @@ RUN git clone -b ${CUPY_VERSION} --recursive https://github.com/cupy/cupy cupy &
194194

195195
# Installing PyTorch
196196
# https://github.com/dusty-nv/jetson-containers/blob/master/scripts/docker_build_ml.sh
197+
#RUN cd /tmp \
198+
# && wget -q --no-check-certificate https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl -O torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl \
199+
# && pip3 install torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl \
200+
# && rm -rf /root/.cache/pip torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl
201+
# https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048
197202
RUN cd /tmp \
198-
&& wget -q --no-check-certificate https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl -O torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl \
199-
&& pip3 install torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl \
200-
&& rm -rf /root/.cache/pip torch-1.11.0a0+17540c5-cp36-cp36m-linux_aarch64.whl
203+
&& wget -q --no-check-certificate https://nvidia.box.com/shared/static/fjtbno0vpo676a25cgvuqc1wty0fkkg6.whl -O torch-1.10.0-cp36-cp36m-linux_aarch64.whl \
204+
&& pip3 install --no-cache-dir torch-1.10.0-cp36-cp36m-linux_aarch64.whl
201205
RUN python3 -c "import torch"
202206

203207
# Download OpenCV

0 commit comments

Comments
 (0)