新买的笔记本,从头开始装环境。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/设备信息说明没问题