赞
踩
官网网址,点击图中download即可自动下载
cmd
回车>>出现如下界面。nvidia-smi
,查看本机所支持的最高cuda版本(记住这个版本号,安装cuda不可以超过这个版本,否则会出问题,驱动程序号同理),如图本机支持的最高cuda版本为11.7,驱动程序版本为516.40.出现如下界面:
cmd
回车>>出现如下界面nvcc -V
查看是否安装成功,出现cuda版本即为成功。1.根据cudnn官网查看cuda与cudnn版本对应关系(!!!很重要,不可以乱下载,下载时需要一个NVIDA账户,用邮箱注册一个即可。11.x或12.x即为对应数字开头的版本都能适配,除了版本对应,cudnn选择较稳定的版本,例如我下载的为11.5.0版本的cuda,此处选择2022年对应11.x版本的cudnn,不追求最新。本人所要安装的是windows系统,因此下面选择windows版本的安装包。
2.解压压缩包,复制文件夹到cuda文件夹中,会自动对应,cuda默认位置为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5,
3.若出现这一提示,继续即可
1.在网上搜索tensorflow和cuda的对应关系,此电脑安装tensorflow_gpu==2.9.0版本。
win+R打开cmd, 创建tensorflow环境,安装也需要在所创建的环境中,可参考b站视频(【有幸遇见-b站最全最简洁易学的深度学习环境配置教程】 https://www.bilibili.com/video/BV1ov41137Z8/?p=7&share_source=copy_web&vd_source=5d48ca2d6d56e40f0372f613cb602850)进行,下面为文字过程。
activate
,进入base环境conda create -n tf python==3.8
,回车,等待安装完成conda activate tf
,进入创建的tf环境pip install tensorflow_gpu==2.9.0 -i https://pypi.douban.com/simple
可自行修改镜像源链接
conda env list
列出本机conda创建的虚拟环境所在位置,如图,找到tf所在位置,在pycharm中选择该环境进行代码测试。
from tensorflow.python.client import device_lib
# 列出所有的本地机器设备
local_device_protos = device_lib.list_local_devices()
# 打印
# print(local_device_protos)
# 只打印GPU设备
[print(x) for x in local_device_protos if x.device_type == 'GPU']
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。