Windows11 + anaconda + tensorflow-gpu

新买的笔记本,从头开始装环境。Windows11+anaconda+tensorflow-gpu
型号:16英寸 R7-7840H 32G 1T RTX4050 120HZ

需要说明的是,注意版本对应关系!

1.查看nvidia-smi信息


命令行输入nvidia-smi,查看CUDA版本为12.0,表示不能超过这个版本

2.确定版本


根据经验,windows不支持2.11及以上的tensorflow-gpu,因此确定使用版本对应关系为:
Python 3.9 Tensorflow-gpu 2.7.0 CUDA11.5.2 cuDNN 8.3.2

3.下载安装CUDA 11.5.2


(system那里选11)
安装结束后,检查环境变量
输入nvcc -V,返回相关信息说明安装成功

4.下载安装cuDNN


安装结束后,配置环境变量,检查环境变量是否成功

对于CUDA和cnDNN的安装网上已经很成熟了,可以查看https://blog.csdn.net/weixin_43412762/article/details/129824339

5.设置新建环境位置


告别C盘,放在D盘(可搞可不搞,看需求)

6.新建环境


conda create -n gputensor python=3.9

7.激活环境


conda activate gputensor

8.安装tensorflow-gpu


因为安装anaconda时自定义的安装位置,所以需要在命令行中cd到安装路径,并确保此时仍然处于新建立并激活的虚拟环境中
pip install tensorflow-gpu==2.7.0 -i https://pypi.mirrors.ustc.edu.cn/simple
验证是否成功:python—import tensorflow as tf—tf. version
显示2.7.0表明安装成功

9.验证是否安装成功


命令行中输入python,并依次输入下列内容
import tensorflow as tf
tf.test.is_built_with_cuda() 返回True说明没问题
tf.config.list_physical_devices(‘GPU’) 返回设备信息说明能用GPU
tf.test.is_gpu_available() 返回True/设备信息说明没问题

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值