文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

2023最新pytorch安装(超详细版)

2023-09-07 14:53

关注

前言


因为我之前安装环境花费了不少时间,也是网上搜了很多,想着自己写一个帮助大家快速安装,少走弯路,快速开启深度之路。

一、判断是否有Nvidia(英伟达显卡)

下面我以我的台式电脑windows10系统为例
第一步:右键开始菜单,在弹出选项中选择任务管理器
在这里插入图片描述

进来就是这个页面,这台是没有GPU的

在这里插入图片描述

这台是有GPU的

在这里插入图片描述

首先点性能选项,然后点GPU。在右上方会显示GPU名称,只有带NVIDIA的英伟达显卡的电脑才能安装GPU版本,否则其他的就只能安装CPU版本。我会写两部分,各位根据自己的电脑选择观看就好

二、CPU版

2.1 安装Anaconda

首先去官网下载Anaconda。
在这里插入图片描述

在这里插入图片描述
注意到路径这块,最好是全英文。
接下来根据个人喜好勾选,可以不用管直接下一步安装好就行。
那怎么验证是否安装好了呢?
在这里插入图片描述
在开始菜单下,找到带有绿色图标的Anaconda,点击就可以打开Anaconda的图形化界面
在这里插入图片描述
点击环境,看到base(root),这个就是默认的一个虚拟环境,代表我们安装成功了。

2.2 创建虚拟环境

为什么需要学会这个呢?其实刚刚安装好Anaconda会自带一个基础环境。但是我们后续的项目每一个需要的安装包不同,为了避免冲突,所以我们可以为每一个项目配置一个虚拟环境,这样就不相互打扰了。

在这里插入图片描述
上面两个是Anaconda命令窗口的位置,随便点击那个都可以。

conda env list

输入上面的命令就可以查看当前有多少虚拟环境。
在这里插入图片描述
• 利用 conda create 指令创建新的虚拟环境

• conda create –n 虚拟环境名字 python=版本

演示如下:
在这里插入图片描述

会出现如下图
在这里插入图片描述
输入y就行,然后就是等待安装。好了以后用指令检查一下,如下,就是成功的。
在这里插入图片描述

2.3安装pytorch

等等,安装前这里还有一个问题,因为我们的安装指令都是从国外的通道下载,下载速度会非常慢。
所以我们想要下载快还是得用国内的镜像地址。

conda config --show

输入上面的指令可以查看我们的通道地址
在这里插入图片描述
在这里插入图片描述
现在我们的环境中通道是defaults,也就是说它会从国外给我下载包。
也可以通过下面的指令看到

conda config --get

在这里插入图片描述
下面是一些镜像地址

镜像名用于创建环境镜像地址
清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
北京外国语大学镜像https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
阿里巴巴镜像http://mirrors.aliyun.com/anaconda/pkgs/main

关于加通道地址这块,推荐是下载时可以指定安装通道,而不是直接去改配置文件,这样出现bug的几率比较小。下面我也会提供这种方法不过不推荐

conda config --add channels 通道地址

那么如何验证是否加上呢?
还是用上面的get指令

在这里插入图片描述
这里也会标注,优先使用我们新加上的通道,如果新通道没有,再使用默认通道去找。

conda config --remove channels 通道地址
conda activate yolov8

在这里插入图片描述

进入pytorch官网

在这里插入图片描述

conda install pytorch torchvision torchaudio cpuonly -c pytorch//这种就是直接从国外下载了

下面这种是镜像加速版

conda install pytorch torchvision torchaudio cpuonly –c 镜像地址

以清华镜像源为例:
在这里插入图片描述
出现下图,输入y即可
在这里插入图片描述
出现下图就是安装好了
在这里插入图片描述

2.4 验证pytorch是否安装成功

  1. 激活对应的虚拟环境(你安装Pytorch的虚拟环境)
    conda activate 虚拟环境名
    在这里插入图片描述

  2. 输入conda list,看有没有pytorch或者torch
    比如
    比如我这里就是pytorch

  3. 输入python
    在这里插入图片描述

  4. 输入 import torch
    在这里插入图片描述

  5. 输入 torch.cuda.is_available(),这个命令是检查我们电脑是否可以用GPU,因为没有,当然就是false。

  6. 如果显示False,就说明我们这个PyTorch安装成功了
    在这里插入图片描述

三、GPU版

3.1 安装Anaconda

在这里插入图片描述

在这里插入图片描述

注意到路径这块,最好是全英文。
接下来根据个人喜好勾选,可以不用管直接下一步安装好就行。
那怎么验证是否安装好了呢?

在这里插入图片描述
在开始菜单下,找到带有绿色图标的Anaconda,点击就可以打开Anaconda的图形化界面
在这里插入图片描述

在这里插入图片描述
点击环境,看到base(root),这个就是默认的一个虚拟环境,代表我们安装成功了。

3.2 创建虚拟环境

为什么需要学会这个呢?其实刚刚安装好Anaconda会自带一个基础环境。但是我们后续的项目每一个需要的安装包不同,为了避免冲突,所以我们可以为每一个项目配置一个虚拟环境,这样就不相互打扰了。

在这里插入图片描述
上面两个是Anaconda命令窗口的位置,随便点击那个都可以。

conda env list

输入上面的命令就可以查看当前有多少虚拟环境。
在这里插入图片描述

• 利用 conda create 指令创建新的虚拟环境

• conda create –n 虚拟环境名字 python=版本

演示如下:
在这里插入图片描述

会出现如下图
在这里插入图片描述
输入y就行,然后就是等待安装。好了以后用指令检查一下,如下,就是成功的。
在这里插入图片描述

3.3 CUDA的准备工作

要想使用GPU得先了解自己电脑的GPU。
在这里插入图片描述
鼠标右键开始菜单,点击任务管理器,进入性能中查看自己电脑GPU型号。

首先安装最新的显卡驱动,进入官网选择适合自己电脑的配置,我的配置如下。
在这里插入图片描述
下载完直接安装就行。

然后是查看自己电脑驱动的版本,win+r,输入cmd,再输入下面的命令

nvidia-smi

![在这里插入图片描述](https://img-blog.csdnimg.cn/e019cba08ca848828bbea6c0c07bbb17.png
我的CUDA Driver版本就是11.6,表示的是驱动所能支持的最大运行API版本就是11.6。我如果要安装CUDA Runtime Version(运行版本),要保证CUDA Driver 版本 >= CUDA Runtime 版本,也就是11.6及以前的。
其实装了Anaconda之后Anaconda会提供一个cudatoolkit工具包,同样包含了CUDA的运行API,可以用来替代官方CUDA的CUDA Toolkit。查看官方CUDA的运行API版本。

nvcc -V

在这里插入图片描述

可以看到是11.1,也就是说我的电脑安装pytorch时最好选择11.1-11.6版本的cuda。

3.4 安装pytorch

等等,安装前这里还有一个问题,因为我们的安装指令都是从国外的通道下载,下载速度会非常慢。
所以我们想要下载快还是得用国内的镜像地址。

conda config --show

输入上面的指令可以查看我们的通道地址
在这里插入图片描述
在这里插入图片描述
现在我们的环境中通道是defaults,也就是说它会从国外给我下载包。
也可以通过下面的指令看到

conda config --get

在这里插入图片描述
下面是一些镜像地址

镜像名用于创建环境镜像地址
清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
北京外国语大学镜像https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
阿里巴巴镜像http://mirrors.aliyun.com/anaconda/pkgs/main

添加通道

conda config --add channels 通道地址

那么如何验证是否加上呢?
还是用上面的get指令

在这里插入图片描述
这里也会标注,优先使用我们新加上的通道,如果新通道没有,再使用默认通道去找。

万一你想删除刚刚的通道怎么办呢?如下

conda config --remove channels 通道地址

关于加通道地址这块,推荐是下载时可以指定安装通道,而不是直接去改配置文件,这样出现bug的几率比较小。不推荐直接去改配置文件这种方法。

安装pytorch前想清楚一个问题,想把pytorch安装到那个环境里。
默认是在基础环境下(base),比如我现在想安装到yolov8这个虚拟环镜

conda activate yolov8

输入上面命令进入想安装pytorch的虚拟环境,上面就是想在yolov8这个虚拟环境中安装pytorch
在这里插入图片描述

进入pytorch官网

3.4.1 conda安装法(不推荐)

在这里插入图片描述
进来发现cuda不符,就去以前的版本找。

在这里插入图片描述
大家根据自己的环境需求找,我最后选择这个。如果网络还可以,就用官网的下载就行。

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

下面介绍一种镜像的下载。
第一步:因为cudatoolkit下载很慢,所以也单独指定一个镜像

Conda install cudatoolkit=版本 –c 镜像地址
镜像名用于创建环境镜像地址 - cudatoolkit
清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
北京外国语大学镜像https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
阿里巴巴镜像http://mirrors.aliyun.com/anaconda/pkgs/main

在这里插入图片描述

第二步:

conda install pytorch torchvision torchaudio  -c 镜像地址
镜像名镜像地址 – pytorch, torchvision, torchaudio
清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
北京外国语大学镜像https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/
阿里巴巴镜像http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
南京大学镜像https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/

在这里插入图片描述

这样后续就好了。

3.4.2 pip安装法(推荐)

进入pytorch官网

在这里插入图片描述
如果你的电脑可以装11.7或者11.8就直接粘贴复制到anaconda窗口,不行就像我一样去以前的版本找。
在这里插入图片描述
在这里插入图片描述

3.5 验证pytorch是否安装成功

  1. 激活对应的虚拟环境(你安装Pytorch的虚拟环境)
    conda activate 虚拟环境名
    在这里插入图片描述

  2. 输入pip list或者conda list,看有没有pytorch或者torch
    在这里插入图片描述

比如我这里就是torch
3. 输入python
4. 输入 import torch
6. 输入 torch.cuda.is_available(),这个命令是检查我们pytorch的GPU能否用。
7. 如果显示True,就说明我们这个PyTorch安装成功了
在这里插入图片描述

四、 pycharm安装与配置

首先进入pycharm官网

在这里插入图片描述
点击download

在这里插入图片描述

下载社区版就够用了。

在这里插入图片描述
在这里插入图片描述
注意路径全英文。

在这里插入图片描述

我只勾了.py,其他的根据自己的需要来勾选。直接下一步等待安装完成。
装完后在开始菜单下,找到pycharm点开。

在这里插入图片描述
我喜欢亮一点,在这里调背景色。

在这里插入图片描述
开始新建项目

在这里插入图片描述
自己指定项目的位置。

在这里插入图片描述
因为我们前面已经配好了环境,直接用我们配好的,点击add去找我们配置的位置

在这里插入图片描述
从左往右按顺序点击,找到自己anaconda安装的位置。
在这里插入图片描述
用我们已经创建好的环境,比如我这里是yolov8

在这里插入图片描述
在这里插入图片描述
点击create,项目就创建好了。

在这里插入图片描述
创建一个文件测试一下

import torchprint(torch.cuda.is_available())

在这里插入图片描述

鼠标右健后,点击run就行
在这里插入图片描述

这就代表成功了,注意CPU版本这里应该是False。

来源地址:https://blog.csdn.net/weixin_44752340/article/details/130542629

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯