【发布时间】:2020-12-23 04:33:03
【问题描述】:
当我在新的 venv 中运行 conda install tensorflow-gpu 时,conda 会给出它将安装的软件包列表,但没有我期望的 CUDNN 或 cudatoolkit 软件包,而其他人也有。我期待像这个问题一样的输出(Is it still necessary to install CUDA before using the conda tensorflow-gpu package?
) 但我得到了这个:
编辑:我可以通过运行这个命令来让它工作
conda create -n tf-gpu python=3.7 anaconda 后跟
conda install tensorflow-gpu
完成此操作后,它自行安装了所有 cudnn 库,并且我能够使用 GPU 运行 tensorflow。但是,由于某种原因,它正在安装 tensorflow 1.14,而我需要 2.x,所以有人知道这是为什么吗?
【问题讨论】:
-
This answer in a previous question会解决你的问题。
标签: tensorflow anaconda gpu