Tôi đã theo liên kết này đây để tải xuống tensorflow với hỗ trợ gpu trên hệ điều hành Ubuntu 20.04 của tôi. Đây là một ảnh chụp nhanh của tôi .bashrc mà tôi nghi ngờ có thể là vấn đề nhưng tôi không thể xác minh điều này.
# >>> khởi tạo conda >>>
#!! Nội dung trong khối này được quản lý bởi 'conda init'!!
__conda_setup="$('/home/weiss/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
nếu [ $? -eq 0 ]; sau đó
đánh giá "$__conda_setup"
khác
nếu [ -f "/home/weiss/anaconda3/etc/profile.d/conda.sh" ]; sau đó
. "/home/weiss/anaconda3/etc/profile.d/conda.sh"
khác
xuất PATH="/home/weiss/anaconda3/bin:$PATH"
fi
fi
bỏ đặt __conda_setup
# <<< khởi tạo conda <<<
xuất LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
xuất PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
xuất LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/local/cuda-11.0/include:$LD_LIBRARY_PATH
echo 'xuất LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
echo 'xuất LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH' >> ~/.bashrc
xuất LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH
xuất LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH
Khi tôi chạy lệnh này
tf.config.list_physical_devices('GPU')
tôi nhận được như sau
2021-09-15 18:11:46.561864: Tôi tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:937] nút NUMA thành công được đọc từ SysFS có giá trị âm (-1), nhưng phải có ít nhất một nút NUMA, do đó, việc trả lại NUMA nút không
15-09-2021 18:11:46.564618: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Không thể tải thư viện động 'libcussolver.so.11'; dlerror: libcussolver.so.11: không thể mở tệp đối tượng được chia sẻ: Không có tệp hoặc thư mục như vậy; LD_LIBRARY_PATH: /usr/lib/cuda/include:/usr/lib/cuda/lib64:/usr/local/cuda-11.0/include:/usr/local/cuda-11.0/lib64:/usr/local/cuda/lib64 :/usr/local/cuda-11.0/lib64:/usr/lib/cuda/include:/usr/lib/cuda/lib64:
15-09-2021 18:11:46.565065: W tensorflow/core/common_runtime/gpu/gpu_device.cc:1835] Không thể mở một số thư viện GPU. Vui lòng đảm bảo rằng các thư viện còn thiếu được đề cập ở trên được cài đặt đúng cách nếu bạn muốn sử dụng GPU. Làm theo hướng dẫn tại https://www.tensorflow.org/install/gpu để biết cách tải xuống và thiết lập các thư viện cần thiết cho nền tảng của bạn.
Bỏ qua đăng ký thiết bị GPU...
[]
Tôi đã thử một số cách thiết lập tiềm năng khác nhưng không có cách nào hiệu quả. Bất kỳ trợ giúp hoặc đề xuất được đánh giá cao.
Đây cũng là một ảnh chụp nhanh của nvidia-smi
+--------------------------------------------- ----------------------------- +
| Phiên bản trình điều khiển NVIDIA-SMI 460.91.03: 460.91.03 Phiên bản CUDA: 11.2 |
|-------------------------------------+----------------- -----+----------------------+
| Tên GPU Persistence-M| Bus-Id Disp.A | Dễ bay hơi Uncorr. ECC |
| Fan Temp Perf Pwr:Sử dụng/Giới hạn| Sử dụng bộ nhớ | GPU-Util Compute M. |
| | | MIG M. |
|=================================================== =====+======================|
| 0 GeForce GTX 1070 Tắt | 00000000:01:00.0 Bật | Không áp dụng |
| 0% 53C P0 34W/200W | 1175MiB / 8085MiB | 2% mặc định |
| | | Không áp dụng |
+------------------------------------+----------------- -----+----------------------+
+--------------------------------------------- ----------------------------- +
| Quy trình: |
| GPU GI CI Loại PID Tên quy trình Bộ nhớ GPU |
| ID ID Sử dụng |
|================================================= ============================|
| 0 N/A N/A 934 G /usr/lib/xorg/Xorg 198MiB |
| 0 N/A N/A 1542 G /usr/lib/xorg/Xorg 515MiB |
| 0 Không áp dụng Không áp dụng 1674 G /usr/bin/gnome-shell 227MiB |
| 0 Không áp dụng Không áp dụng 2843 G /usr/lib/firefox/firefox 219MiB |
+--------------------------------------------- ----------------------------- +
