File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff 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
197202RUN 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
201205RUN python3 -c "import torch"
202206
203207# Download OpenCV
You can’t perform that action at this time.
0 commit comments