WIN10+RTX 2080Ti深度学习平台搭建
前置声明:博文中有一些图片来自网上截图(因为安装完成后做的记录,有些安装过程图嫌麻烦没有复现),侵删。
可选步骤:
如果之前有python环境,需要卸载掉全部依赖包,进而再次重新安装的话,那需要知道如何一次性卸载掉全部原有python依赖包。操作方法如下:
第一步:打开CMD,随便进入一个目录下(这里我进入的是C:\Program Files\Python38\Scripts),然后执行下面的命令(txt文件名自己取):
pip freeze > requirements.txt
pip uninstall -y -r requirements.txt
这里,-y的意思是默认全部同意,这样就不用一直输入y了。
下面进入正题:
1. 查看tensorflow、cuda、cudnn等各安装包的版本对应关系
(1)进入https://tensorflow.google.cn/install/source_windows查看tensorflow和cuda,cudnn的对应关系。截至当前截图如下:
(2)打开NVIDIA控制面板,查看你的显卡支持的CUDA驱动(要>=你想安装上方条目中的CUDA版本)
看到这,你应该就能决定出自己应该安装哪个版本的安装包。这里我安装tensorflow版本信息图里面第一个条目。