如何从anaconda安装tensorflow-gpu?

xa9qqrwz  于 8个月前  发布在  其他
关注(0)|答案(1)|浏览(100)

我已经运行了一些非常基本的步骤(tensorflow-gpu目前是2.12.1):

conda create --name py311_tf212 python=3.11 numpy numba scipy spyder pandas
conda activate py311_tf212
time conda install -c conda-forge tensorflow-gpu

经过3个小时的思考和打印几千行软件包依赖项后,安装失败了。
我的系统特点是:

  • Ubuntu 18.04
  • 功能:/Linux-64::__glibc==2.27,
  • CUDA 11.8,
  • Nvidia驱动程序520.61.05.

我不知道还有什么相关的信息。我很乐意得到任何提示。

mepcadol

mepcadol1#

老实说,如果你已经尝试了很多东西,也许最好是重新安装你的ubuntu,重新开始?👀
无论如何
1.首先需要安装nvidia驱动,可以通过软件和更新安装:

只要选择一个与**(专有,测试)**,在您的情况下,它可能会更高,如535或什么。您可以通过运行以下命令来检查是否安装了nvidia-drivers:

nvidia-smi

1.安装Anaconda(运行以下命令):

wget https://repo.anaconda.com/archive/Anaconda3-2023.09-0-Linux-x86_64.sh
shasum -a 256 Anaconda3-2023.09-0-Linux-x86_64.sh
bash ~/Downloads/Anaconda3-2023.09-0-Linux-x86_64.sh

Source for installing Anaconda
1.创建conda环境并通过运行以下命令激活它:

conda create -n any_name python=3.9
conda activate any_name

1.运行以下注解以安装最新的pip版本:

pip install --upgrade pip

1.最后,运行这个:

# For installing tensorflow-gpu
pip install tensorflow[and-cuda]

Source for installing tensorflow-gpu
你可以通过在编辑器上运行以下命令来检查tensorflow是否安装并使用gpu:

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

搞定了!

相关问题