目录

硬件配置及安装

相关软件安装

背景

2020年疫情让我们在家中,这让我开始为自己动手配置一台深度学习主机。因为并没有什么钱,但是对性能还是有一定要求的,最终选择了8000元左右的装机配置。

硬件配置

所有硬件和价格计算以京东价参考并附上了相关链接

时间:2020年初

所有硬件总价格:¥8845.1

不算键盘、鼠标、显示器价格:¥7983.3

机箱风扇:先马(SAMA)游戏风暴 12CM无光 机箱风扇 x2 ¥15.9 x2 = ¥31.8

硬件装机

跟着视频装机是最方便的

硬件茶谈在 bilibili 里的这个视频【装机教程】这可能是你能在网上找到最详细的装机教程 ,对电脑组装讲的非常好,建议装机前看一遍,装的时候也可以跟着走

另外,我因为买的是 CPU 盒装,暂时没买单独 CPU 风扇,Intel CPU 自带风扇安装,请看这里 从零开始学装机 教你如何安装CPU风扇

系统安装

Windows 10 系统安装,和上面一样推荐硬件茶谈在 bilibili 的视频 超详细WIN10系统安装教程

这里我不建议用PE,直接系统安装就好了,并不麻烦,而且我心里感觉这样安装一直比PE好一些的

显卡驱动安装/更新

因为 Win10 一般会自动为我们安装一个驱动,所以这里我们一般更新一下

笔记本需要选 notebooks,其他的按照自己的显卡型号

CUDA Toolkit 安装

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题(百度百科)。

简单来说 CUDA 是帮助我们的深度学习程序访问和利用 GPU 计算的一个框架

cuDNN 安装

cuDNN 是 CUDA toolkit 中对于深度神经网络的另外一个库,但是我们需要手动把他们加入 CUDA 中

下载解压后,将 bin, include, lib 中的文件收到拷贝到前面安装 cuda toolkit 的对应文件夹中

创建 Python 环境

创建环境 conda create -n pytorch pip python=3.7

安装 PyTorch

进入刚刚创建的 conda 环境 conda activate pytorch

安装 PyTorch conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

安装 PyCharm IDE

新建 project,配置 existing environmnet 为刚刚创建的 conda 环境 ,路径大致为 Anaconda3\env\pytorch\python.exe(这里需要找到,anaconda的目录,然后找env,最后找到刚刚装机的环境名字为pytorch里的 python 作为解释器)

测试 pytorch 和 cuda

任意进入 python 环境

# 测试 pytorch

import torch

x = torch.rand(5,3)

print(x) # 有一个5x3 的矩阵输出

# 测试cuda

torch.cuda.is_available() # 期待结果为 True

参考