Skip to content

Commit

Permalink
Remove duplicated cuDNN copy
Browse files Browse the repository at this point in the history
  • Loading branch information
zifeitong committed Jul 24, 2024
1 parent da4b411 commit 90a9566
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions cuda-devpack.Containerfile
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,10 @@ EOF
RUN <<EOF
wget -nv --show-progress --progress=dot:mega ${CUDNN_URL}${CUDNN_ARCHIVE}.tar.xz
tar -xf ${CUDNN_ARCHIVE}.tar.xz
cp -r ${CUDNN_ARCHIVE}/include/* /usr/local/cuda/include
cp -r ${CUDNN_ARCHIVE}/lib/* /usr/local/cuda/lib64
cp -r ${CUDNN_ARCHIVE}/LICENSE /usr/local/cuda/LICENSE.cudnn
mv ${CUDNN_ARCHIVE}/include/* /usr/local/cuda/include
mv ${CUDNN_ARCHIVE}/lib/* /usr/local/cuda/lib64
mv ${CUDNN_ARCHIVE}/LICENSE /usr/local/cuda/LICENSE.cudnn
rm -r ${CUDNN_ARCHIVE}
rm ${CUDNN_ARCHIVE}.tar.xz
EOF

Expand Down

0 comments on commit 90a9566

Please sign in to comment.