Windows 下安装CUDA和Pytorch跑深度学习

整体的安装顺序是：CUDA -> Anaconda -> Pytorch

Step1：CUDA 安装

注意：cuda安装的版本需要根据pytorch的版本走，否则你pytorch建立的网络模型在不对应的cuda上是跑不起来的！！！cuda版本装错的话还要卸载，挺麻烦的；

1、查看Pytorch官网最新版本支持的cuda版本

官网： PyTorch

点击Install

可以看到我们需要下载cuda11.6版本

2、去cuda官网下载对应版本

官网：CUDA Toolkit Archive | NVIDIA Developer

随便选择一个下载。

打开网址有点慢，但是下载还挺快的；

3、安装CUDA

双击安装包，一路默认下去就行；

4、验证安装是否成功

快捷键win+R输入cmd打开命令行窗口输入：nvcc --version

安装成功！yeah！

Step2：安装Anaconda

这个没啥要求，这里主要讲一下安完之后创建一个新的环境；

1、查看pytorch需要的python版本

anaconda中的新环境（我们需要在这个环境中跑深度学习）中的python版本也需要对应pytorch版本进行安装；在本文第二张图的位置往下翻就能看到了；

2、创建新环境

打开Anaconda Powershell Prompt命令行窗口（找不到就直接在搜索中搜索），刚进去是base环境

conda create --name my_pytorch python=3.7   # 创建一个python版本为3.7的名字为my_pytorch的环境 
conda env list  # 查看当前所有环境
conda activate my_pytorch  # 激活刚才新建的my_pytorch环境

Step3:安装Pytorch

在Anaconda Powershell Prompt命令行窗口激活我们新建的my_pytorch环境之后,输入安装pytorch包的命令

等待，就安装完成了

Step4：测试

Windows 下安装 CUDA 和 Pytorch 跑深度学习 - 动手学深度学习v2_哔哩哔哩_bilibili

在命令行里输入nvidia-smi可以查看GPU使用率；

注：【CUDA】nvcc和nvidia-smi显示的版本不一致？ - 简书 (jianshu.com)