今天尝试一下GPU在机器学习中的使用。
公司里的办公电脑,看品牌基本都是名牌,以联想为主、还有很多HP之类。
说起显卡,却没有啥像样的, 基本都是些集成显卡。
远不如人家网吧打游戏的电脑,人家只是玩玩游戏而已,而我们吃的是草,还想下奶。
不管怎么说,自己像李云龙一样,总算搞了一个独显。
依次打开-控制面板–大图标–NVIDIA控制面板–系统信息,打开看看。
Nvidia GeForce GT 710显卡,驱动版本471.41,有1G内存,192个cuda核心。
虽然很菜,但也已是最好了,骑兵连 总算有底子了。
驱动程序是Nvidia官方网站 https://www.nvidia.cn/ 下的,
然后,按照网友的介绍,开始一步步开始安装。
【CUDA】cuda安装 (windows版)
https://blog.csdn.net/weixin_43848614/article/details/117221384#t0
我装的是Cuda11.4+cuDNN11.4
然后是装tensorflow,为加快速度使用了科大镜像
pip install tensorflow -i https://pypi.mirrors.ustc.edu.cn/simple/
也有人说,得专门装一下Tensorflow的GPU版
pip install tensorflow-gpu
(貌似装不装这一步,都没啥影响)
然后,python程序测试GPU
import tensorflow as tf
a = tf.constant(1.)
b = tf.constant(2.)
print