搭建TensorFlow-GPU环境
安装anaconda
创建一个虚拟环境,起名如TF
激活该环境
在该环境中安装TensorFlow -gpu版本
在该环境中使用命令
conda install cudatoolkit=10.1
conda install cudnn=7.6
不需要单独到NVIDIA官网下载驱动!!!
相关视频教程:
https://www.bilibili.com/video/BV1Cg4y1q7Xq?p=8
有用的话点个赞!
pip install -i https://pypi.doubanio.com/simple tensorflow-gpu==2.1.0
conda install scikit-learn