【基于windows和Anaconda,安装gpu版pytorch】

目录

一.检查显卡与python版本

桌面鼠标右键进入NVIDIA控制面板--帮助--系统信息--组件--查看cuda版本为11.3

​

二.创建pytorch环境

进入 Anaconda prompt 命令窗口，输入

三.安装 PyTorch

在 Anaconda prompt 窗口中，复制官网提供的命令并输入(若想使用镜像源加速，需要去掉末尾的-c pytorch)

四.测试

---

一.检查显卡与python版本

桌面鼠标右键进入NVIDIA控制面板--帮助--系统信息--组件--查看cuda版本为11.3

---

二.创建pytorch环境

进入 Anaconda prompt 命令窗口，输入

conda create -n pytorch python=3.8

1. 进入pytorch官网：https://pytorch.org/

2. 可看到下图，官网会自动根据你的电脑，显示你可安装的CUDA版本，并给出安装命令

3. 激活pytorch环境，输入以下代码

   activate pytorch

---

三.安装 PyTorch

在 Anaconda prompt 窗口中，复制官网提供的命令并输入(若想使用镜像源加速，需要去掉末尾的-c pytorch)

conda install pytorch torchvision torchaudio cudatoolkit=11.3

---

四.测试

输入python，进入python开发环境中

import torch
torch.cuda.is_available()

 看到True, 终于成功