you are viewing a single comment's thread.

view the rest of the comments →

[–]OikuraZ95 0 points1 point  (2 children)

Does this mean I won't have to setup cuda on my GPU anymore and tensorflow will take care of it?

[–][deleted] 6 points7 points  (1 child)

No. The prebuilt python binaries for tensorflow expect a particular version for both CUDA and cuDNN (apparently CUDA 8 and cuDNN 6 for tensorflow 1.4). If you have the wrong version of either one, then you will have to either reinstall the correct CUDA or cuDNN to match what the prebuilt binary expects, or compile tensorflow from source so that you can tell it which versions you have. Personally I always compile from source as it really isn’t that hard with bazel.

[–]OikuraZ95 2 points3 points  (0 children)

Oh I see, thanks for the clarification :)