环境:linux ubuntu
-
安装anaconda及配置
直接去官网下载,地址:https://www.anaconda.com/distribution/#download-section
下载好了之后安装
bash Anaconda.sh
创建tf虚拟环境
conda create -n tf python=3.7
查看所有环境
conda env list
激活tf环境
conda activate tf
安装tensorflow2 cudnn cuda 等
conda install tensorflow-gpu==2.0
-
安装pycharm
下载,解压,安装
添加桌面快捷方式sudo gedit /usr/share/applications/Pycharm.desktop
输入
[Desktop Entry]
Type=Application
Name=Pycharm
GenericName=Pycharm3
Comment=Pycharm3:The Python IDE
Exec=sh /home/chu/pycharm-2019.3.1/bin/pycharm.sh //修改
Icon=/home/chu/pycharm-2019.3.1/bin/pycharm.png //修改
Terminal=pycharm
Categories=Pycharm; -
安装显卡驱动
查看可以安装的驱动
ubuntu-drivers devices
自动安装推荐驱动
sudo ubuntu-drivers autoinstall
重启
sudo reboot
检查安装结果
nvidia-smi
4. 测试结果
在pycharm创建项目,选择刚才创建的tf环境
然后粘贴代码测试
import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())
结束