文章目录

  • ​​1. 创建和激活新的环境​​
  • ​​2. 安装pytorch前的准备​​
  • ​​2.1 查看设备型号​​
  • ​​2.3 安装cuda​​
  • ​​3. 安装pytorch​​
  • ​​4. 测试​​

视频连接：​​https://www.bilibili.com/video/BV1hE411t7RN?p=1​​

1. 创建和激活新的环境

conda create -n pytorch python=3.6
conda activate pytorch

2. 安装pytorch前的准备

2.1 查看设备型号

安装pytorch之前需要安装cuda，调用gpu加速，所以需要知道gpu的型号

1、查看显卡型号

方法一：打开设备管理器->显示配适器，查询到为NVIDIA GeForce GTX 1050

win10安装pytorch_python


方法二：打开任务管理器->性能

win10安装pytorch_python_02



2、查看驱动版本

打开NVIDIA控制面板->帮助->系统信息

win10安装pytorch_python_03


查看cuda适配版本

win10安装pytorch_设备型号_04


根据驱动版本选择对应的cuda版本

win10安装pytorch_设备型号_05



2.3 安装cuda

cuda官网​​链接​​，翻进去傻瓜式安装

老版本的cuda链接：​​链接​​

win10安装pytorch_python_06

输入​​nvcc -V​​查看是否安装成功

win10安装pytorch_设备型号_07



3. 安装pytorch

4、根据步骤1、2查询到的信息，进入pytorch官网，​​https://pytorch.org/​​，选择对应版本，复制指令，进入anaconda对应环境下输入指令安装pytorch

先换源，下载的快

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/

后复制下面指令，注意去掉-c pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2

win10安装pytorch_python_08


报错：

win10安装pytorch_python_09


看别人博客​​链接​​，说是因为anaconda不是默认装在C盘才报错

解决方法步骤：找到anaconda文件夹右击打开属性->安全->修改为system

win10安装pytorch_设备型号_10


再一次运行安装指令，之后便安装成功。

win10安装pytorch_python_11

4. 测试

打开pycharm或者就在终端上测试：

import torch
torch.cuda.is_available()

返回为True

win10安装pytorch_python_12


win10安装pytorch_官网_13