Tensorflowgpu与Anaconda的安装

2024-04-26 00:52:05 发布

您现在位置:Python中文网/ 问答频道 /正文

本周末,我一直在尝试安装Tensorflow,并在我的计算机上使用GPU支持,但我在使用pip/conda方面不是很有经验,在观看和尝试了许多不同的教程/方法后,我现在感到非常困惑

我有一个GeForce GTX 1650图形卡,我已经安装了Cuda 10.0(也是11.2,但我从“PATH”中删除了它,并且只使用10.0版本,我认为这不是问题)

我已经为CUDA 10下载了cuDNN 7.5.0,我认为我已经正确复制和放置了文件(安装了cuDNN)

我只是想让一些版本的Tensorflow gpu工作,但是你可以在图片上看到我一直在尝试的Tensorflow版本

我曾尝试从我的计算机上安装和卸载Python(我也多次重新安装Anaconda),因为如果我在Anaconda环境中安装Python版本(在我的示例中为Python 3.7),我不确定是否需要安装Python版本(在我的系统上)

是否有人知道如何使用我的设置(cuDNN 7.5.0、CUDA 10)在Windows 10上安装Tensorflow GPU,或者可能在Python版本中遇到过一些问题或类似于我的Anaconda问题?

enter image description here


Tags: pip方法版本gputensorflow计算机教程anaconda
1条回答
网友
1楼 · 发布于 2024-04-26 00:52:05

按照以下步骤在windows系统上安装Tensorflow GPU

  1. 确保安装了正确版本的Visual studio。勾选here

  2. 按照上述说明here为windows系统设置CUDA

  3. 安装Tensorflow

#check current python version
python  version
#Create the virtual environment
conda create -n tf python=PYTHON_VERSION 
#Activate the tf environment
conda activate tf
#Install Tensorflow
pip install tensorflow
#Install CUDA and cuDNN using conda and make sure CUDA and cuDNN version should match the Tensorflow version
conda install -c anaconda cudatoolkit=10.0 cudnn=7.5

相关问题 更多 >