tensorflow-gpu 安裝

Cold_Chair發表於2024-07-05

不知道為啥 MT 為啥還有人在用 tensorflow 這種古老的東西,搞得被迫赤石。

https://tensorflow.google.cn/install/source_windows?hl=zh-cn#gpu 翻到最下面,看看 tensorflow-gpu 的版本和 CUDA,python 的版本的對應關係。

然後比如

pip install tensorflow-gpu==2.6.0 

就可以了,有其他問題再補。

相關文章