使用Conda的Tensorflow 1.15 + CUDA + cuDNN安装

jdzmm42g 于 2023-03-13 发布在其他

关注(0)|答案(3)|浏览(205)

我正在尝试安装tensorflow-gpu 1.15使用Conda的CUDA和cuDNN的一个简单的安装。问题是，检查官方网站的兼容性图表，我需要python 3.6，CUDA 10.0和cuDNN 7.4。
通过conda search cudnn搜索Conda rep，结果显示没有cuDNN 7.4。有没有其他方法来安装所需的软件包？或者tensorflow 1.15也可以与其他版本组合使用？
顺便说一句，python 3.6，tensorflow-gpu 1.15和CUDA 10安装正确，但似乎没有cuDNN我就不能正确使用GPU。我最近才开始使用Conda，所以也许有一个直接的方法来做到这一点，我没有意识到。我的Conda版本是4.9.1（迷你版）。
---更新---
以防我在尝试conda create -n myenv -c conda-forge tensorflow-gpu=1.15时添加错误：

Collecting package metadata (current_repodata.json): done
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.
Collecting package metadata (repodata.json): done
Solving environment: - 
Found conflicts! Looking for incompatible packages.
This can take several minutes.  Press CTRL-C to abort.
failed                                                                                                                                                                                                                                        

UnsatisfiableError: The following specifications were found to be incompatible with each other:

Output in format: Requested package -> Available versions

Package _tflow_select conflicts for:
_tflow_select==2.1.0=gpu
tensorflow==1.15.0 -> _tflow_select[version='2.1.0|2.3.0|2.2.0',build='gpu|mkl|eigen']
Note that strict channel priority may have removed packages required for satisfiability.

tensorflow

来源：https://stackoverflow.com/questions/64811841/tensorflow-1-15-cuda-cudnn-installation-using-conda

3条答案

按热度按时间

hujrc8aj1#

我不确定这是否是问题所在，但我是通过以下方式安装的

conda create -n tensorflow1.15 python=3.5
conda activate tensorflow1.15
conda install cudatoolkit=10.0
conda install cudnn=7.3.1
pip3 install tensorflow-gpu==1.15

而且它看起来和GPU配合得很好。我不知道cuDNN 7.3.1和7.4一样工作。最好的方法是用conda安装tensorflow，但是它给予了我一个错误，试图安装tensorflow-gpu=2.X。
也许有趣的是，你可以用conda search -c nvidia <packageName>搜索CUDA和类似的官方安装程序。

赞(0）回复(0）举报 2023-03-13

sdnqo3pr2#

我会让conda自己处理所有依赖项，通过conda而不是pip安装tensorflow。GPU版本的tensorflow可以在流行的conda-forge频道中获得：

conda create -n myenv -c conda-forge tensorflow-gpu=1.15

赞(0）回复(0）举报 2023-03-13

dvtswwa33#

TensorFlow 1.15的最佳设置是遵循以下指南：https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/tensorflow-1.14/install.html#tf-install。
推荐的版本为CUDA 10.0和cudNN 7.6.5

注意将要安装的protobuf版本，如果您执行gpu版本，它是protobuf 4.21.1，但您必须使用以下命令重写它：

pip install --upgrade tensorflow-gpu==1.15 "protobuf<4.0"
如果您使用CPU版本，建议在此处使用此版本：（https://github.com/protocolbuffers/protobuf/releases/tag/v3.4.0）以避免错误。只需下载protoc-3.4.0-win32.zip（windows）
希望能有所帮助。

赞(0）回复(0）举报 2023-03-13

我来回答

使用Conda的Tensorflow 1.15 + CUDA + cuDNN安装

3条答案

相关问题

热门标签

最新问答