Python环境—tentorflow\cuda\cudnn安装

1 电脑查显卡型号

任务管理器-性能-GPU

 2 nvidia官网下载显卡型号对应的cuda

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

3 nvidia官网下载cuda对应的cudnn

https://developer.nvidia.com/rdp/cudnn-download

4 新建虚拟环境-安装tensorflow-gpu

tensorflow-gpu是tensorflow和其他gpu部分库的合集

需要核对tensorflow-gpu和cudnn、cuda版本对应

pip install tensorflow-gpu=2.3.1 -i https://pypi.doubanio.com/simple

5 在anaconda里检测gpu是否可用:

import tensorflow as tf
tf.config.list_physical_devices('GPU')

在检测过程中遇到,缺各种dll,在自己电脑里搜索找找、网上找找,复制到nvidia安装路径下:

网上可用的dll网盘地址:

tensorflow-gpu版本使用时报错,各种dll文件找不到_布布汪的博客-CSDN博客

 6 It's OK!


上述文章部分内容可能源于网络

整理在此,只为个人记录学习过程,并分享给需要的小伙伴

如有侵权,联系即删

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值