Pytorch 安装记录（GPU）

一、CUDA安装

CUDA安装如下博客：

(50条消息) 【CUDA】cuda安装 （windows版）_haoweixl的博客-CSDN博客_cuda安装

前提是电脑有NVIDIA独显 

在电脑桌面上 点击鼠标右键，出现下图，点击NVIDIA控制面板

点击系统信息，查看显卡型号，驱动程序版本

                               

 二、pytorch安装

用清华镜像源安装的方法

我在anacaonda里面用清华镜像源安装老是不对，另行他路，参考博客

1、用anaconda 创建一个环境

Win+R启动cmd，在命令提示符内输入以下命令：

conda create –n pytorch python=3.7
activate pytorch

这里一共存在两条指令：
前面一条指令用于创建一个名为pytorch的环境，该环境的python版本为3.7。
后面一条指令用于激活一个名为pytorch的环境。
2、打开pytorch的官方安装方法：

里头有推荐安装的torch 和torchvision版本

我的是CUDA11.1， 推荐安装pytorch==1.8.0 torchvision==0.9.0

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

下载whl:https://download.pytorch.org/whl/torch_stable.html

 cu111表示CUDA的版本是11.1， 若要下载cpu版的，则选择前面是cpu的。

cp38表示python3.8,  记住要安装torch, 再装torchvision。

下载完成后的安装路径：（用的别人的图）

 在cmd定位过来后利用文件全名进行安装就行了

3、检验是否安装成功

先输入python，然后输入import torch，如果输入后没有任何报错，没有任何显示那就是成功了，然后再输入torch.cuda.is_available()，返回的是True，那便是完成了整个操作。