1 电脑查显卡型号
任务管理器-性能-GPU
2 nvidia官网下载显卡型号对应的cuda
3 nvidia官网下载cuda对应的cudnn
https://developer.nvidia.com/rdp/cudnn-download
4 新建虚拟环境-安装tensorflow-gpu
tensorflow-gpu是tensorflow和其他gpu部分库的合集
需要核对tensorflow-gpu和cudnn、cuda版本对应
pip install tensorflow-gpu=2.3.1 -i https://pypi.doubanio.com/simple
5 在anaconda里检测gpu是否可用:
import tensorflow as tf
tf.config.list_physical_devices('GPU')
在检测过程中遇到,缺各种dll,在自己电脑里搜索找找、网上找找,复制到nvidia安装路径下:
网上可用的dll网盘地址:
tensorflow-gpu版本使用时报错,各种dll文件找不到_布布汪的博客-优快云博客
6 It's OK!
上述文章部分内容可能源于网络
整理在此,只为个人记录学习过程,并分享给需要的小伙伴
如有侵权,联系即删