深度学习环境搭建(Windows10)
安装Anaconda
进入Anaconda官网,下载windows版本的安装文件。
下载好后一路点击Next保持默认就行,最后的安装路径可选择在D盘,因为后期的环境可能会比较大,放在C盘个人感觉不是很好。
安装后,打开菜单栏,打开Anaconda。
这里有Anaconda PowerShell Prompt 和 Anaconda Prompt 两个选项,我们平常选择 Anaconda Prompt 打开即可。
打开conda面板后,输入conda --version,如果能成功输出即表示安装成功。
安装CUDA
在桌面空白处右键,选择“NVIDIV 控制面板”,打开后点击“系统信息”。如果右键没有这个选项的话,百度一下解决。
选择“组件”,查看CUDA的版本号。
这里可以看到,CUDA是11.4的版本,所以在安装CUDA时不能比这个版本高。为了能得到最高的算力,需要选择一个离这个版本最贴近的CUDA安装,我这里安装的是11.3的版本。(最好看一下pytorch官网,pytorch的版本需要跟cuda的版本对应,如果装了11.4的CUDA,但是pytorch还没有11.4对应的版本,那岂不是还得自己去编译,都是坑。)
打开CUDA官网,选择适合自己版本的CUDA下载。
如果推荐的版本号不是自己需要的版本号,可以去旧版本中找。
双击cuda的exe文件,选择自定义安装。
去掉Visual Studio Integration的勾选。
如果本机的驱动版本(当前版本)小于cuda对应的版本(新版本),则选择,否则不选。如果勾选错误会导致安装失败,以及电脑频繁蓝屏死机。
记住安装位置,tensorflow要求配置环境,后面要用,最好是截图留着备用。
等待安装完成,查看环境变量。
测试一下cuda是否安装成功。
win+R打开命令窗口,输入cmd。
打开dos窗口后,输入nvcc --version,回车。
有以下输出就算是成功了。
安装CUDNN
cudnn的版本需要与刚才cuda的版本相对应,否则会导致GPU加速不能使用。
进入cudnn官网,需要注册或登录一下。
以本文档为例,CUDA选择了11.4的版本,那么CUDNN必须要选择对应的版本,如果网页推荐没有,那么就去旧版本中查找。
因为没有找到恰好对应11.3CUDA的版本,我选择了对应CUDA11.x的版本。
下载后解压,把解压后的内容放至CUDA的安装目录下即可,这个路径可以从安装CUDA步骤中找到。
安装Pytorch
打开conda。
用conda创建一个新的python为3.8.5的环境,以免影响到全局环境。
conda create -n Python3.8.5-Pytorch python=3.8.5
conda activate
打开pytorch的官网,点击”Install“,根据CUDA的版本,选择合适的pytorch版本。
最好是不要选择自己去编译最新版的,可能会出现各种问题,有点坑。
如果没有找到自己需要的版本,就去老版本中找。
我这里恰好有11.3对应的版本,直接根据要求安装就好。
复制这一行代码,到conda的环境下粘贴运行。等运行完也就安装好了,如果安装失败,重新再运行一次,conda会接续上一次安装失败的断点继续安装。
安装完成后,测试一下是否安装成功。
依旧是在conda的环境下,输入python。进入python命令行。
输入import torch,如果不报错,即说明pytorch安装成功。
再判断一下CUDA是否可用。
继续输入print(torch.device(‘cuda’) if torch.cuda.is_available() else torch.device(‘cpu’))
如果打印CUDA即代表cuda安装成功,否则说明安装失败,不可用GPU加速。
安装Pycharm
pycharm可用来写python文件,训练神经网络模型。
进入Pycharm下载地址,他有Professional专业版和Community社区版两种,我们装社区版Community的就可以了,因为它免费。
点击下载。
下载完成后,双击即可安装,改一下安装地址即可。
安装成功后打开Pycharm,随便打开一个项目,点击“File>>Settings”。
settings页面,左侧栏选择“Project>>Python Interpreter”,右侧的下拉框中选择“Show All”。
我们添加一下,刚刚再conda中新建的python环境。
添加环境页面,左侧栏中选择“Conda Environment”,右侧中选择“Existing environment”,并在Interpreter中,选择刚刚创建的python.exe。
如此环境便算搭建好了,我们写一个test.py文件尝试一下。
右键选择创建Python File文件,输入文件名test.py。
写入代码后右键点击运行。
成功!