tensorflow安装——GPU版

tensorflow安装——GPU版

安装CUDA

1. 先查看电脑的cuda版本
   
   版本对应
   在版本对应网址里查看对应的cuda、cudnn、python、tensorflow_gpu版本。
   如果cuda版本是11.6的话，上面网址没有对应版本，可以按照
   cuda 11.6
   cudnn 8.9.0
   python 3.8
   tensorflow_gpu 2.10.0
   来安装，亲测没问题。cuda下载地址
   下载完成后直接安装即可，注意第一个路径是解压缩路径，安装完成后会自动删除，第二个路径才是安装路径（建议默认），需记住。用于配置系统环境变量。
   具体流程：
   • 选择自定义版本：
     
   • 选择安装路径
     
   • 如果电脑提示说找不到的话取消选择visual studio
     

配置环境变量

• 系统环境变量
  
• 用户环境变量
  

判断是否安装成功： nvcc --version。

安装cudnn

cudnn下载地址
cudnn并不是软件，只是作为cuda的一个补充。下载对应版本的cuda并安装完成后，将下载的cudnn的三个文件夹复制到cuda的安装路径里面。
cudnn文件

cuda文件夹

安装tensorflow

需要预先安装好anaconda环境，打开 anaconda Prompt ，激活想要安装tensorflow_gpu的虚拟环境，输入

pip install tensorflow-gpu==2.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

不要直接复制安装命令，要注意版本号要相互对应。

是否安装成功

import  tensorflow as tf
tf.test.is_gpu_available()

值为True则说明安装成功。

cuda安装参考：链接