「安装」 Windows下安装CUDA和Pytorch

发布于:2024-10-12 ⋅ 阅读:(197) ⋅ 点赞:(0)

「安装」 Windows下安装CUDA和Pytorch

Mac、Linux、云端

Mac、Linux、云端安装Miniconda和Pytorch的方法参考其他资料。

Windows

下面进行Windows下安装CUDA和Pytorch来跑深度学习。

安装CUDA

CUDA介绍:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

注意只有NVIDIA GPU才能使用CUDA,AMD GPU或集成显卡均无法使用CUDA。

首先查看电脑是否拥有NVIDIA GPU:
win + r 打开「运行」窗口,输入dxdiag
在这里插入图片描述
在这里插入图片描述

下载CUDA,进入官网 https://developer.nvidia.com/cuda-downloads
在这里插入图片描述

打开安装包:
在这里插入图片描述
设置安装目录:
在这里插入图片描述
安装选默认选项:
核心就是装:

  1. 驱动
  2. cuda开发环境

在这里插入图片描述

安装完成后进入命令行,使用nvidia-smi命令查看是否安装成功:
在这里插入图片描述

安装miniconda

接下来下载Python环境miniconda(为什么要装miniconda参考其他资料),进入官网 https://docs.anaconda.com/miniconda/miniconda-other-installer-links/#
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
默认选项安装即可:
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

在开始菜单打开 Anaconda Powershell Prompt:
在这里插入图片描述
输入python --version,查看是否是自己安装的python版本:
在这里插入图片描述

创建虚拟环境

安装完conda后默认是base环境,最好创建一个自己的虚拟环境,因为实际中都是在服务器中运行的,每个人都要使用自己的虚拟环境。使用conda创建虚拟环境的方法见:博客

输入此条指令创建conda虚拟环境,可以自己更改名称和python版本:

conda create -n 自己输入名称 python=版本号

如下所示,示例创建名为test的环境

conda create -n test python=3.8

在这里插入图片描述
输入下行代码激活刚才创建的环境,主要名称要与之前创建时设置的名称相应:

conda activate 名称

例如我们输入conda activate test激活刚才创建的虚拟环境:
在这里插入图片描述

注意,这里运行后,左边括号里变成你虚拟变量的名称,即为进入此环境了,之后你输入的所有命令都是在此环境中执行的。

需要在虚拟环境中重新安装包如jupyter,d2l:
在这里插入图片描述

同样需要在虚拟环境中安装PyTorch(由于之前在base环境中装过PyTorch,再在虚拟环境中安装是很快的):
在这里插入图片描述

conda的一些其他指令

退出虚拟环境

输入conda deactivate当左边的小括号变为base了,即为退出此虚拟环境了:
在这里插入图片描述

环境重命名

conda rename -n test test1,前一个test为旧名字,后一个test1为新名字,注意要在base环境中运行此代码,更名后按新名字启动环境。

查找所有创建过的环境

conda env list此代码可查找到你创建过的所有环境。带*的为当前所属环境。
在这里插入图片描述

查看当前环境下安装的包

conda list

conda list -n 名称

在这里插入图片描述

删除环境

删除虚拟环境:

conda remove -n 名称 --all

conda env remove --name 名称

安装PyTorch

接下来安装PyTorch的GPU版本,进入官网 https://pytorch.org/
在这里插入图片描述
选择好对应的版本,然后将对应pip命令复制到Anaconda Powershell Prompt中进行pytorch安装:
在这里插入图片描述
整个过程还是比较慢的,文件也比较大,下载速度很慢。

下面看下是否安装成功,在Anaconda Powershell Prompt中打开python,导入torch:
在这里插入图片描述

测试

安装好后找一个真正的深度学习的应用跑一下,在《动手学深度学习》官网 https://zh-v2.d2l.ai/ 下载Jupyter记事本:
在这里插入图片描述
下载好之后解压(解压的过程提示有重名文件,选择覆盖):
在这里插入图片描述

接下来使用pip命令安装所用到的包 jupyter、d2l。打开Anaconda Powershell Prompt,使用pip命令进行安装:

pip install jupyter d2l

(d2l中包含要用到的包如numpy、pandas。安装d2l时就会将这些包一并安装,如果以后要用到其他包可以再单独安装)

最好使用源镜像进行安装,否则容易出现安装失败。
在这里插入图片描述
在这里插入图片描述

安装好上面两个包之后在Anaconda Powershell Prompt中启动jupyter notebook就可以运行jupyter实例了:
在这里插入图片描述
启动jupyter notebook后会直接跳出浏览器窗口:
在这里插入图片描述

在跳出的浏览器窗口中并没有找到前面下载的jupyter实例。问题出现在刚才是在 C:\Users\Mao 目录下启动jupyter notebook的,而下载的jupyter实例是在 E:\ 目录下,因此需要使用 cd 命令切换到 E:\ 目录下启动jupyter notebook

注意:
启动jupyter notebook的目录需要与jupyter实例(d2l-zh)所在目录一致。

在这里插入图片描述

切换到jupyter实例目录下再启动jupyter notebook:
在这里插入图片描述
这样在跳出的浏览器窗口中就出现下载的jupyter实例(d2l-zh)了:
在这里插入图片描述
依次点击 d2l-zh、pytorch、chapter_convolutional-modern、resnet.ipynb,这就是要测试的实例:
在这里插入图片描述
点击Kernel - Run All 运行所有:
在这里插入图片描述
点击Restart:
在这里插入图片描述
然后拉到页面的最下面,就可以看到已经开始下载数据进行训练了:
在这里插入图片描述

在训练的过程中可以通过任务管理器或在Anaconda Powershell Prompt中输入nvidia-smi命令查看GPU使用率:
在这里插入图片描述

训练结果:
在这里插入图片描述
作为对比,看一下在云上的处理速度(去《动手学深度学习》中找到现代卷积神经网络 - 残差网络,便可找到数据):
在这里插入图片描述

总结

在Windows上学习深度学习的步骤:

  1. 确认有NVIDIA GPU
  2. 安装CUDA
  3. 安装miniconda
  4. 安装GPU版PyTorch
  5. 安装d2l和Jupyter
  6. 下载d2l记事本运行测试

其他

还可以使用谷歌Colab(免费算力平台)进行深度学习,colab有免费的CPU和GPU可以使用。

使用colab方法:代码选择PyTorch版本
在这里插入图片描述
然后拉到页面最上面,点击colab[PyTorch]:
在这里插入图片描述
在这里插入图片描述
不过使用谷歌需要一点方法。


网站公告

今日签到

点亮在社区的每一天
去签到