tensorflow-gpu==2.0 tensorflow2.0 gpu版安装教程 for windows and linux

anaconda 安装方法。(windows,linux版本通用)

1,安装anaconda
2,建立一个虚拟环境并指定python版本
conda create -n myenv python3.6
3,切换到新建到环境
conda activate myenv
3,默认安装最新tensorflow
conda install tensorflow
#这时安装的是1.14,因为conda里面只有这个版本。等安装完后,我们再手动安装
pip install tensorflow-gpu==2.0

即可完成安装。genda安装时会自动安装cuda 和cudnn。

自己安装

不通过conda安装就有点麻烦了
需要先去官网下载cuda,不需要注册。
再去下载cudnn。下载cudnn后把解压的文件放在cuda的bin同级目录 起名为cudnn即可
注意版本问题

liux手动安装 比较特别:
首先要注意的是 tensorflow2 只支持cuda10.0 不支持10.1 千万记住

给电脑装显卡驱动:
1,关掉securtiy boot mode 先设置密码,然后disable
去英伟达官网下驱动: https://www.geforce.cn/drivers

2,sudo apt-get purge nvidia*
#nouveau是某些linux默认的显卡驱动,因为要安装新的驱动所以必须先让这个失效并删除
3,sudo apt-get --purge remove xserver-xorg-video-nouveau
sudo apt-cache search nvidia | grep -E “nvidia-[0-9]{3}”
sudo apt-get update
sudo apt-get install gcc make
#安装新下的显卡驱动
chmod +x NVIDIA-Linux-x86_64-418.56.run
sudo ./NVIDIA-Linux-x86_64-418.56.run
#重启
sudo shutdown -r now
#如果安装成功能看到显卡信息
nvidia-smi

你可能感兴趣的:(#,tensorflow,#,python)