conda安装tensorflow GPU版本
时间: 2023-11-25 13:04:40 浏览: 128
安装GPU版本的Tensorflow
安装TensorFlow GPU版本需要确保你的计算机上已经安装了支持CUDA和cuDNN的NVIDIA显卡驱动程序。以下是安装步骤:
1. 安装CUDA Toolkit
在官网上下载并安装与你的GPU版本相对应的CUDA Toolkit。可以在 https://developer.nvidia.com/cuda-downloads 页面找到适合你的版本。
2. 安装cuDNN
在官网上下载并安装与你的CUDA版本相对应的cuDNN。可以在 https://developer.nvidia.com/cudnn 页面找到适合你的版本。
3. 创建conda虚拟环境
在命令行窗口中输入以下命令来创建一个名为“tensorflow-gpu”的虚拟环境:
```
conda create --name tensorflow-gpu
```
4. 激活虚拟环境
在命令行窗口中输入以下命令来激活虚拟环境:
```
conda activate tensorflow-gpu
```
5. 安装TensorFlow GPU版本
在命令行窗口中输入以下命令来安装TensorFlow GPU版本:
```
conda install tensorflow-gpu
```
6. 验证安装
在命令行窗口中输入以下代码来验证TensorFlow GPU版本是否安装成功:
```python
import tensorflow as tf
print(tf.__version__)
```
如果成功安装,输出的版本号应该是TensorFlow GPU版本。
阅读全文