CUDA और CuDNN संस्करण Tensorflow2.4.1 के विरुद्ध संघर्ष करते हैं

आप सभी को धन्यवाद। मैंने पाया कि मेरी समस्या नीचे की तरह हल हो गई है।

Ubuntu 20.04 स्थापित होने के बाद, नवीनतम NVIDIA ड्राइवर स्थापित करें। (शायद कोई ज़रूरत नहीं है, क्योंकि यह CUDA स्थापना के साथ स्वचालित रूप से स्थापित हो जाएगा।)

sudo ubuntu-drivers devices
sudo apt-get install --no-install-recommends nvidia-driver-460
nvidia-smi
sudo reboot

CUDA11.0 स्थापित करें जो TF2.4.1 चाहता है। इस बार मैं डेब [स्थानीय] के बजाय रनफाइल [स्थानीय] का उल्लेख करता हूं जो मुझे पहले विफलता मिली थी। https://डेवलपर। nvidia.com/cuda-11.0-update1-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=runfilelocal

sudo apt update
wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
sudo sh cuda_11.0.3_450.51.06_linux.run

*चेतावनी 'ड्राइवर का मौजूदा पैकेज मैनेजर इंस्टालेशन मिला।' ऊपर की लाइन पर दिखाई दिया और मैंने इसे निरस्त कर दिया। इसके बजाय नीचे किया गया।

sudo sh ./cuda_11.0.3_450.51.06_linux.run --toolkit --silent –-override

पुष्टि करें कि क्या usr/local/cuda-11.0/bin मौजूद है और संपादित करें bashrc.

sudo vim ~/.bashrc

Bashrc में निम्नलिखित 2 पंक्तियाँ जोड़ें।

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

पुनः लोड करें

source ~/.bashrc

CuDNN स्थापित करें। टैर्ड लाइब्रेरी को पहले से डाउनलोड कर लें। https://developer.nvidia.com/rdp/cudnn-archive

cd Downloads
tar -xzvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
sudo apt update

TF2.4.1 को https://www.tensorflow.org/install/pip?hl के रूप में इंस्टॉल करें =ja निर्देश देता है।

sudo apt install python3-dev python3-pip python3-venv
cd Documents/ML/tf-test1
python3 -m venv test1
source test1/bin/activate
pip install --upgrade pip
pip list
pip install --upgrade tensorflow

जांचें कि क्या टीएफ ठीक से स्थापित है।

import tensorflow as tf
print(tf.__version__)
print(tf.config.list_physical_devices('GPU'))
print(tf.test.gpu_device_name())

जब मैं अपना सीएनएन मॉडल करता हूं, 'कनवल्शन एल्गोरिदम प्राप्त करने में विफल। यह शायद इसलिए है क्योंकि cuDNN इनिशियलाइज़ करने में विफल रहा' फिट के रूप में हुआ ()। फिर मैंने हमेशा की तरह नीचे किया।

export TF_FORCE_GPU_ALLOW_GROWTH=true

फिर सुलझा लिया।