mirror of
https://github.com/gosticks/openpose.git
synced 2026-08-12 12:20:22 +00:00
14 lines
618 B
Bash
Executable File
14 lines
618 B
Bash
Executable File
#!/bin/bash
|
|
|
|
echo "This script assumes Ubuntu 16 or 14 and Nvidia Graphics card up to 10XX. Otherwise, it will fail."
|
|
|
|
# Install cuDNN 5.1
|
|
if [[ $UBUNTU_VERSION == *"14."* ]] || [[ $UBUNTU_VERSION == *"15."* ]] || [[ $UBUNTU_VERSION == *"16."* ]]; then
|
|
CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
|
|
wget -c ${CUDNN_URL}
|
|
sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
|
|
rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig
|
|
else
|
|
echo "cuDNN NOT INSTALLED! Ubuntu 16 or 14 not found. Install cuDNN manually from 'https://developer.nvidia.com/cudnn'."
|
|
fi
|