3060Ti显卡,tensorflow2.4-GPU安装
⼀、写在前⾯:
本来⼀直⽤的tensorflow2.3,换了新电脑就原样装的,但是怎么都调⽤不了GPU,2.3版本的安装⼤家可以参考:
因为之前经受过很多折磨,所以⽐较快的想到可能是显卡,tensorflow,cuda的适配问题,事实证明确实,所以⼤家如果之后遇到类似的问题在⽐较确认⾃⼰安装流程没有问题的情况下,优先考虑这个问题。
⼆、安装步骤:
电脑配置:cpu:11400H;显卡:3060Ti(显存8G),系统:win10
1、安装anaconda
火炬之光2 联机平台2、安装vscode
3、配置虚拟环境:
conda create -n tf2.4 python=3.8
activate tf2.4
4、安装cudatoolkit和cudnn:
⾸先确认⾃⼰显卡适配的cuda版本:
nvidia-smi
李曼铱
可以看到cuda的版本是11.4,其实,安装逻辑是根据这个版本确定要⽤哪个版本的tensorflow,我的1060ti显卡的电脑,cuda版本是10,所以⽤tf2.3。
conda install cudatoolkit=11.0
conda install cudnn==8.2.1
家长感谢老师的话还有⽐较多的教程中cudnn是在官⽹下载安装的,也是可以的,anaconda中库的管理和更新可能会有点慢,不过这个⽅法⽬前亲测有效,之后的版本如果不⾏,就,再说吧。查看版本:
conda search cudnn
田启文个人简历
5、安装tensorflow2.4
pip install tensorflow-gpu==2.4.0 -i mirrors.aliyun/pypi/simple/ 6、查看是否正确安装
蒙面歌王 普罗米修斯conda list
python
import tensorflow as tf
tf.__version__
李茂鼎
有如下显⽰说明安装正确: