Anaconda安装与TensorFlow GPU版配置
GPU版本 NVIDIA DeForce GTX1050 Max-Q
支持CUDA
Anaconda安装
打开Anaconda官网,选择对应版本下载安装
安装时勾选 Add Anaconda3 to my PATH environment variable
安装完成后打开,进入主界面
点击 Environment 中的 Create 创建新环境,这里选择的Python 3.8.0
至此,Anaconda环境配置告一段落
CUDA11.4安装
网址 CUDA Toolkit 11.4 Update 2 Downloads | NVIDIA Developer
下载完成后安装即可
cuDNN8.2.0 for CUDA 11.4安装
下载完成后获得的是一个压缩包,解压后得到三个文件夹
找到CUDA的安装路径
只需要把cuDNN各文件夹中的文件复制到CUDA同名文件夹下即可
验证安装
通过NVIDIA提供的 deviceQuery.exe 和 bandwidthTest.exe 来查看GPU的状态,两者均在安装目录的 extras\demo_suite文件夹中
首先运行 deviceQuery.exe ,可以看到如下结果:
运行 bandwidthTest.exe 看到如下结果:
如果出现命令行闪屏可以先打开命令行,然后拖入exe文件解决
安装TensorFlow GPU 版
打开Anaconda,启动刚才创建的环境,启动Terminal
使用pip安装TensorFlow GPU
pip installtensorflow-gpu==2.6.0
安装完成后可以进去python环境进行测试
importtensorflowastf
tf.__version__
tf.test.is_gpu_available()
TensorFlow GPU版安装完成,版本为2.6.0
安装jupyterlab
在Terminal中使用pip安装jupyterlab
pip install jupyterlab
安装完成后,启动jupyterlab
jupyter lab
之后会自动启动浏览器打开网址
路径则为Terminal当前所在路径