anaconda安装tensorflow gpu
时间: 2023-04-30 09:03:11 浏览: 151
要在Anaconda中安装TensorFlow GPU版本,需要执行以下步骤:
1. 首先,确保您已经正确安装了NVIDIA驱动程序和CUDA工具包
2. 创建一个新的conda环境,并在其中安装TensorFlow GPU版本,可以运行命令 `conda create --name tf-gpu tensorflow-gpu`
3. 激活新环境,运行 `conda activate tf-gpu`
4. 最后检查是否正确安装,运行 `python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"`
如果你在上述过程中有遇到问题,或者你安装的tensorflow不是最新版,可以尝试更新一下,在安装的环境中运行 `conda update --all`
阅读全文