2017-04-02 24 views
6

Chúng tôi có cần cụ thể cuDNN v5.1 (như gợi ý) cho TensorFlow hay phiên bản mới nhất (v6.0) có hoạt động không? Có khả năng tương thích ngược trong các phiên bản cuDNN không?CuDNN v6.0 có hoạt động với TensorFlow không?

+2

Phiên bản mới nhất (1.0.1 TensorFlow) đã không làm việc với cuDNN v6 cho tôi (tính đến tháng 1).. – Ali

+0

Ah, cảm ơn bạn. Sau đó, các phiên bản cuDNN có khả năng không tương thích ngược và với mỗi phiên bản mới, chúng ta cần phải đợi cho bất kỳ cơ sở hạ tầng nào được nâng cấp/tinh chỉnh cho phiên bản mới ... – hekimgil

Trả lời

2

Không, cuDNN 6.0 không được hỗ trợ trong phiên bản 1.2 mới nhất. Nhưng có một hy vọng: chính thức release notes nói như sau:

TensorFlow 1.2 có thể là lần cuối cùng chúng tôi xây dựng với cuDNN 5.1. Bắt đầu từ với TensorFlow 1.3, chúng tôi sẽ cố gắng tạo tất cả các tệp nhị phân dựng sẵn của chúng tôi với cuDNN 6.0. Trong khi chúng tôi sẽ cố gắng giữ mã nguồn tương thích với cuDNN 5.1, nó sẽ là nỗ lực tốt nhất.

Vì vậy, hy vọng phiên bản 1.3 tiếp theo sẽ sử dụng cuDNN 6.0. Đặc biệt là bây giờ, khi 7.0 is right around the corner.

1

Đây sẽ là câu trả lời được cập nhật của phản hồi của Salvador Dali.

Tôi đã nâng cấp phiên bản tensorflow thành 1.2.1 và sau đó cudnn 6.0 dường như không hoạt động. Tôi đã sử dụng pip

pip install tensorflow 
pip install --upgrade tensorflow 

Chi tiết cài đặt là here.

2

Hôm nay (ngày 21 tháng 8 2017) Tôi đã cài đặt mới nhất phát hành v1.3 Tensorflow và tôi có thể xác nhận rằng nó YÊU CẦU cuDNN v6.0 và không hoạt động với v5.1. Nó sẽ yêu cầu thực tế cho thư viện libcudnn.so.6 và không phải là libcudnn.so.5

P.s. Nếu bạn muốn nó hoạt động với cuDNN 5.1. bạn có thể cài đặt phiên bản trước, ví dụ: v1.2 mà là tại liên kết này:

https://www.tensorflow.org/versions/r0.12/get_started/os_setup#virtualenv_installation

Các vấn đề liên quan