深度学习第一步 Windows+NVIDA配置 CUDA pytorch 环境

深度学习第一步 Windows+NVIDA配置 CUDA pytorch 环境

感谢

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

目录

• 深度学习第一步 Windows+NVIDA配置 CUDA pytorch 环境
• 感谢
• 1 查看电脑的GPU类型
• 2 cuda 下载与安装
• 3 安装anaconda
• 4 在anaconda下创建一个环境
  • 点击绿色箭头启动环境
• 5 下载Pytorch
• 6 安装完成，检验

1 查看电脑的GPU类型

1 在 cmd命令行之中 输入dxdiag / 在运行中shuru dxdiag

2 之后就会打开 DirectX 诊断工具

在显示一栏下面就有显卡信息

2 cuda 下载与安装

CUDA Toolkit 11.6 Update 2 Downloads | NVIDIA Developer

安装选默认选项就可以，把驱动和开发环境选上

看一下系统环境配置

在命令行中输入用nvidia-smi来看一下是否安装成功

3 安装anaconda

Anaconda | The World's Most Popular Data Science Platform

4 在anaconda下创建一个环境

点击绿色箭头启动环境

5 下载Pytorch

Start Locally | PyTorch

在激活的环境中输入命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

6 安装完成，检验