tensorflow-gpu 安装
时间: 2023-10-05 17:07:10 浏览: 94
TensorFlow-gpu安装,N卡需求文件
安装tensorflow-gpu需要按照以下步骤进行操作:
1. 首先,使用conda创建一个tensorflow环境,命令为:conda create -n tensorflow python=3.6 (python版本根据需求下载)。
2. 然后,使用activate tensorflow命令进入到之前创建的tensorflow环境。
3. 接下来,使用pip install tensorflow-gpu==2.0.0命令进行tensorflow-gpu的安装。注意,根据之前的cuda和cudnn的版本选择相应的tensorflow-gpu版本进行安装。
4. 最后,安装完成后,可以使用import tensorflow在python中验证tensorflow-gpu是否安装成功。
阅读全文