一、安装CUDA
前阵给电脑刷机,准备重新学一下深度学习,先从比较简单的pytorch框架入手,利用电脑1060显卡加速训练,记录配置环境需要注意的问题。
首先需要确定电脑是支持cuda的,查询本机显卡是否支持CUDA的安装,是否存在列表中,
网址:https://developer.nvidia.com/cuda-legacy-gpus.
如果有下载你想要的cuda版本,我下载的是10.2版。点击进入网址https://developer.nvidia.com/cuda-toolkit-archive.选择想要下载的版本,进行下载。
下载文件为exe可执行文件,点击执行。