操作系统:windows11专业版23H2
语言环境:Anaconda3-2024.02-1-Windows-x86_64.exe
CUDA包:cuda_11.2.2_461.33_win10.exe
CUDNN包:cudnn-11.2-windows-x64-v8.1.1.33.zip
pycharm:pycharm-professional-2024.1.1.exe
硬件:华硕枪神6Plus 英伟达3070Ti 移动版
一,安装Anaconda3
(1)安装软件,官网:https://www.anaconda.com/installation-success?source=installer
选择最新版安装
之后的选项全部打勾,完成安装
(2)测试是否安装成功,cmd命令:
conda -V
出现版本号即安装成功
(3)虚拟环境相关cmd命令:
创建环境:
conda create -n ten_gpu_39 python=3.9
激活环境:
conda activate ten_gpu_39
更换pip源:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
关闭环境:
conda deactivate
删除环境
conda remove -n ten_gpu_39 --all
二,安装CUDA
(1)安装cuda包,观察版本决定是否勾选,只有更新的版本才能安装成功,否则失败,下一步等待安装完成
(2)测试cuda是否安装成功,cmd命令:
nvcc -V
出现这些信息表示安装成功
三,安装cudnn
(1)解压cudnn-11.2-windows-x64-v8.1.1.33.zip文件,主文件夹命名为cudnn,子文件夹含有如图
将cudnn文件放入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2(cuda默认路径)下
(2)添加环境变量
结束,最后conda根据python版本安装tensorflow支持gpu的相应版本,pycharm配置codan环境即可
tensorflow中文官网:https://tensorflow.google.cn/install/pip?hl=zh-cn