关于anaconda 安装 tensorflow gpu问题

一般情况下我会采用ancanda直接进行安装;

原因一:无需为cuda和cudnn另外去下载和配置,ancanda可以在虚拟环境里面为您通过命令的方式直接安装好。

原因二:目前的pytorch和百度的飞桨框架都可以通过conda命令一键安装好环境,包括(cuda和cudnn)非常方便。

tensorflow需要注意的地方:安装tensorflow的时候犹豫没有conda安装命令,貌似官方也没有推荐使用conda安装,这里就会给安装tensorflow带来一定的疑惑。

其实实际上tensorflow是可以使用conda进行安装的,我们可以先创建一个虚拟环境:

conda create -n tf36 python=3.6

然后激活该环境:

conda activate tf36

接下来安装cuda 和cudnn,还是还是使用conda命令安装(因为安装超级方便,无需我们自己手动去下载,而且必须先安装好cuda和cudnn再去安装tensorflow,否则会失败)

conda install cudatoolkit=10.0
conda install cudnn=7.6.5

然后,我们再通过pip命令的方式进行安装tensorflow:(requirements.txt,这个是因为我的项目是已经配置好环境的,直接通过requirements.txt进行安装,如果您不懂requirements.txt请百度)

pip install -r requirements.txt

这样的话,我们就安装好了tensorflow的环境,而且是在虚拟环境里面安装的cuda不会影响我们其他的环境。非常完美!!

 

posted @ 2020-10-09 13:47  likecs  阅读(326)  评论(0编辑  收藏  举报