ฉันจะทำให้ tensorflow gpu ทำงานได้อย่างง่ายดายได้อย่างไร
ฉันจะทำให้ tensorflow gpu ทำงานได้อย่างง่ายดายได้อย่างไร
คำตอบ:
ฉันเพิ่งติดตั้ง Tensorflow GPU บน Ubuntu 18.04 มีคำแนะนำมากมาย แต่ฉันคิดว่าวิธีที่เร็วและง่ายที่สุดมักจะไม่ได้ใช้และฉันต้องการแบ่งปัน:
NVIDIA DRIVER:
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
nvidia-smi
CUDA:
โดยปกติ: "sudo apt install nvidia-cuda-toolkit" อย่างไรก็ตามสิ่งนี้จะทำการติดตั้งเวอร์ชั่น 9.1 ใหม่เกินไปในขณะนี้และจะไม่สามารถใช้งานได้ แทนที่จะดาวน์โหลด CUDA 9.0: https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal แล้วรัน (รายละเอียดเพิ่มเติมเกี่ยวกับสายแรกที่สามารถพบได้ที่: วิธีการสามารถ ฉันติดตั้ง CUDA 9 บน Ubuntu 17.10 ):
sudo sh cuda_9.0.176_384.81_linux.run --override
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
cd ~
sudo nano .bashrc
add at the end of the file:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
CTL+X to save and exit
CuDNN:
ลงทะเบียนที่นักพัฒนา nvidia https://developer.nvidia.com/cudnn ดาวน์โหลด 9.1 runtime & ไลบรารีนักพัฒนาซอฟต์แวร์สำหรับ 16.04 (ไฟล์ cuDNN v7.1.3 Runtime Library สำหรับ Ubuntu16.04 (Deb) & cuDNN v7.1.3 ไลบรารีนักพัฒนาสำหรับ Ubuntu16.04 ( Deb)) เปิดไฟล์ด้วยตัวจัดการซอฟต์แวร์และติดตั้ง ตรวจสอบกับ:
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
ติดตั้งไลบรารีและเทนเซอร์โฟลว์:
sudo apt-get install libcupti-dev
pip3 install tensorflow-gpu
ตรวจสอบ:
in tensorflow check for GPU support
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
หมายเหตุ: เริ่มต้น tensorflow หรือสภาพแวดล้อมการพัฒนาของคุณจากเทอร์มินัลมิฉะนั้นสำหรับฉันมันไม่ได้โหลดตัวแปร PATH
เพียงเพื่อขยายคำตอบของจอนนี่: เมื่อทำตามวิธีของจอนนี่สำหรับการติดตั้ง CUDA ฉันต้องเลือก "ไม่" สำหรับ "ติดตั้งไดร์เวอร์กราฟิก NVIDIA เร่งสำหรับ Linux-x86_64 384.81?" (ใช่หรือเริ่มต้นสำหรับทุกอย่างอื่น) ไม่เช่นนั้นการติดตั้งไดรเวอร์ CUDA จะล้มเหลวในการพูดว่า "ข้อผิดพลาด: โมดูลเคอร์เนล NVIDIA 'nvidia-drm' จะปรากฏขึ้นให้โหลดในเคอร์เนลของคุณแล้ว"
จากนั้นคุณสามารถทดสอบการติดตั้ง CUDA ของคุณโดยการเรียกใช้ตัวอย่าง
ก่อนติดตั้งการอ้างอิงของพวกเขา:
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libglfw3-dev libgles2-mesa-dev
สิ่งเหล่านี้จะไม่รวบรวมกับคอมไพเลอร์ใหม่ใน Ubuntu 18 ดังนั้นคุณจะต้องปรับลดรุ่นเป็น gcc / g ++ 6:
sudo apt-get install gcc-6 g++-6
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-6 10
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 10
รวบรวมตัวอย่างโดยไปที่:
~/NVIDIA_CUDA-9.0_Samples/bin/x86_64/linux/release
และรวบรวมโดยใช้:
make -k
รันหนึ่งในตัวอย่างที่ชอบ:
./nbody