环境
显卡:GeForce MX330
系统:Win10,64位
一、Anaconda3安装
下载,一路next,So easy,不多说!
二、tensorflow-gpu安装
1、第一个问题,版本太多,装哪个?
看看本机支持什么版本,打开naconda命令行走起:conda search --full --name tensorflow
看来支持2.5.0啊,准备搞它!
因为要安装GPU版本的TF,还得看看本机的显卡支持CUDA版本与TF2.5.0是否匹配,我们查一下。
查看显卡支持的CUDA版本,找到 NVIDIA控制面板-帮助-系统信息-组件
根据需求,不超过这个版本!
tensorflow 2.5.0 稳定版介绍推荐“利用 CUDA11.2 和 cuDNN 8.1.0 构建 tensorflow pip 软件包”,CUDA 11.2本机可以支持。
最总确定版本:tensorflow