tensorflow - 可以在同一个 conda 环境中安装 Rapids 0.16 和 TF 2.2 吗？

Question

我已经尝试过并且总是遇到冲突。

标签： tensorflowcondarapids

Answer 1

我运行了以下命令并且能够成功安装 RAPIDS-0.16 和 Tensorflow 和 Tensorflow-gpu 2.2 ：

• 首先，使用https://rapids.ai/start.html上的命令之一在 conda 环境中安装 rapids 。我用了 ： conda create -n rapids-0.16 -c rapidsai -c nvidia -c conda-forge -c defaults rapids=0.16 python=3.7 cudatoolkit=10.2

• 激活 conda 环境：conda activate rapids-0.16

• 然后使用安装 tensorflowpip install tensorflow==2.2

请检查您是否使用了正确的 CUDA 库版本，并检查是否使用正确的 Python 版本来安装 RAPIDS 和 Tensorflow。