一、安装pytorch准备工作

1.   按照步骤安装Anaconda,其中注意要把环境变量添加到path中,安装Anaconda就相当于安装了python,安装成功以后在命令提示符中输入python会出现其版本说明安装成功,如下图所示安装的python版本是3.9.12。

安装Anaconda的另一个好处在于我们可以自己创建一个新的环境,当我们需要使用时直接切换环境运行程序即可。

centos pytorch cpu环境 pytorch配环境_人工智能

2.   安装开发环境,在Anaconda Prompt中输入如下命令,不要看软件,这里我整错了,用的时VScode,如果你用如下方法会很麻烦,之后的命令也都是在Anaconda Prompt中操作。

centos pytorch cpu环境 pytorch配环境_CUDA_02

3.    安装成功之后会提示激活环境的命令(conda activate pytorch)和关闭环境的命令(conda deactivate)。

centos pytorch cpu环境 pytorch配环境_人工智能_03

4.   输入激活pytorch的命令, 激活成功之后会从原始环境切换到pytorch环境下(在Anaconda Prompt中操作)。

centos pytorch cpu环境 pytorch配环境_官网_04

5.   使用pip list 查看工具包会发现并没有我们所需要的pytorch工具包。

centos pytorch cpu环境 pytorch配环境_CUDA_05

二、安装pytorch

1.   进入官网下载pytorch。

centos pytorch cpu环境 pytorch配环境_python_06

2.   准备工作

        (1)判断驱动版本是否符合要求:在命令提示符中输入nvidia-smi命令,出现如下图所示,查看驱动版本且需要大于396.36,小于的话需要用管家将驱动升级。若输入nvidia-smi命令显示不是内部或外部命令可

centos pytorch cpu环境 pytorch配环境_CUDA_07

         (2)基于显卡安装pytorch:开始菜单搜索设备管理器,点击显示适配器查看GPU型号是否是英伟达显卡

                 无英伟达显卡或只有集显,CUDA选择 “None”;

                 有英伟达显卡,CUDA推荐选择9.2,或者11.3;

centos pytorch cpu环境 pytorch配环境_官网_08

将各个选项选好以后在终端输入命令即可,先别输,往下看!!!

centos pytorch cpu环境 pytorch配环境_pytorch_09

经验分享:使用官网命令安装会非常非常慢,方法有如下两种:

方法一:可以将已有的pytorch安装包和cuda放到Anacond安装目录的pkgs文件夹下再去终端或者Anaconda Prompt中执行安装命令。

方法二:方法一太慢的话就使用清华镜像重新下载,因为官网下载的包是从国外服务器下载的所以很慢,而清华镜像是从开源软件的软件包库中同步过来的镜像,在国内用这个镜像可以加快下载速度。使用镜像安装pytorch可按如下网址进行操作。

安装的CUDA版本要小于本机的CUDA版本(二.2图中驱动版本后),不然CUDA会不可用。

centos pytorch cpu环境 pytorch配环境_人工智能_10

centos pytorch cpu环境 pytorch配环境_python_11

 

centos pytorch cpu环境 pytorch配环境_pytorch_12

3.   在pytorch环境下(conda activate pytorch)输入python,再输入如下各个命令并出现指定的信息说明安装成功。

centos pytorch cpu环境 pytorch配环境_CUDA_13

如果不成功,(1)看看自己的VScode使用的解释器是不是pytorch(ctrl+p),选择pytorch的解释器再输入命令,(2)亦或者在Anaconda Prompt中激活pytorch环境后再次输入一下pytorch的安装命令,(3)如果还是不成功出现报错尝试重启电脑,或者关机等一段时间再开机。

import torch 后出现“页面文件太小,无法完成操作”时

centos pytorch cpu环境 pytorch配环境_pytorch_14