Advertisement

TensorFlow GPU版安装指南(Linux系统)

阅读量:

一、TensorFlow-gpu环境的安装

在进行该代码的安装操作之前,需确保系统中已配置好相关的深度学习运行环境

(例如:conda、pytorch、cuda、cudnn等软件环境)

TensorFlow的版本需与所使用的GPU版本相匹配,用户可根据自身需求自行选定相应版本,亦可不指定具体版本直接完成TensorFlow-gpu的安装

复制代码
    conda create -n tf3 python==3.8
    
复制代码
    conda activate tf3
    

当前所配置的CUDA运行环境版本为11.8,Python解释器版本为3.8,同时安装了tensorflow-gpu==2.7.0的软件包。

具体安装步骤如下:

复制代码
    pip install tensorflow-gpu==2.7.0 -i  https://pypi.mirr

全部评论 (0)

还没有任何评论哟~