首先需要准备的东西:Anaconda环境
查看自己GPU型号并安装英伟达驱动程序
首先要知道自己电脑的GPU版本型号:右击我的电脑->管理->设备管理器。
比如我的时NVIDIA GeForce GTX 1060 6GB,
英伟达GeForce系列 ,型号为GTX1060 6GB运行内存,知道这个以后再去官网去看自自己相应的软件驱动
查询自己的GPU所对应的驱动:CUDA GPU | NVIDIA Developer 查询到我的算力时6.1
下载GPU驱动 Download Drivers | NVIDIA
点击search进入下载驱动程序
建议就按照软件的默认路径装,这样可能会把上一把不正确的删除掉,安装完成后的界面如下:
安装cudatoolkit
CUDA与显卡驱动:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
TensorFlow-GPU与CUDA cudnn Python版本关系:https://tensorflow.google.cn/install/source_windows?hl=en#gpu
打开这个 输入命令:
- conda install cudatoolkit=10.1
- conda install cudnn=7.6
- pip install tensorflow-gpu==2.1.2