pytorch的深度学习环境安装配置(GPU版)

这篇具有很好参考价值的文章主要介绍了pytorch的深度学习环境安装配置(GPU版)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

目录

一些概念理解

 0.anaconda配置国内镜像源

 1.anaconda建立一个新的虚拟环境

 2. 更新显卡驱动CUDA Driver

 3. 安装pytorch

3.1 法(一):利用pip安装Pytorch

3.1.1 法(一)在线pip安装

 3.1.2 法(二)本地pip安装

3.2 法(一):利用conda安装Pytorch

3.3 验证pytorch是否安装成功

 4. Pycharm配置虚拟环境

4.1 给新建项目配置虚拟环境

4.2 给下载项目配置虚拟环境 

 参考:



一些概念理解

  • 显卡(GPU):GPU(Graphics Processing Unit),用于屏幕显示图像,用于与视频、图像处理相关的矩阵运算等任务。
  •  驱动:操作系统(OS)的插件。由于GPU是硬件,因此我们需要安装驱动,让计算机识别特定硬件。
  • CUDA:NVIDIA(英伟达)显卡的指令集,驱动程序的扩展包。有了CUDA,我们就可以操作NVIDIA显卡

pytorch的深度学习环境安装配置(GPU版)

        (图片来自厉害的B站up主我是土堆) 

        安装pytorch之前,我们需要先查看电脑是否有英伟达显卡,根据与pytorch版本,是否需要安装更新对应驱动、Cuda。 

  • anaconda:虚拟环境管理软件,可以建立不同的虚拟环境。
  • pytorch/tensorflow:人工智能方面的代码框架体系
  • pycharm/jupyter notebook:python的编辑器,类似记事本,编辑软件,本身不具备代码执行功能。(其中jupyter notebook是anaconda自带的)
  • 深度学习中各个软件关系:

pytorch的深度学习环境安装配置(GPU版)

(图片from B站up主:我是土堆) 

        我们在Pycharm中配置好使用哪个虚拟环境,并写好python的代码后,运行程序,该python文件通过配置找到对应虚拟环境,将python文件交给其中的python解释器。python解释器对该python文件进行解析,在解析过程中从安装的Package中调用需要的库,并将其翻译成计算机语言,发送给CPU,CPU执行对应指令。当CPU发现我们希望一些运算数据转移到GPU上,CPU发送对应指令,通过驱动,将数据发送给GPU进行运算。

参考:(6条消息) 深度学习环境配置中软件之间逻辑与关系_我是土堆的博客-CSDN博客https://tudui.blog.csdn.net/article/details/125728710

0.anaconda配置国内镜像源

         使用anaconda下载各种包时,其默认源,即提供安装包的服务器,是其公司自己设立的,国外服务器,速度慢,不稳定,因此我们可通过以下指令将下载源配置为国内镜像源,以较快速度(起飞)完成下载、安装工作。

        在Anaconda Prompt里复制以下命令行即可。

清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls true

中科大源:
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes

        可使用以下指令,查看镜像源是否配置成功。

conda config --get

pytorch的深度学习环境安装配置(GPU版)

         其中“defaults"是anaconda默认的通道(下载地址),是最低优先级,而我们添加的新的通道(国内镜像地址)是最高优先级。因此,我们下载包时,anaconda优先在添加的国内通道里查找所需要的包,如果没有,再去默认的国外通道查找。

PS: 若要删除镜像加速通道,则把”add channels”改为“remove channels"

conda config --remove channels 【通道地址】


conda config --remove channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --remove channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --remove channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --remove channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --remove channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --remove channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/

1.anaconda建立一个新的虚拟环境

(anaconda 建立一个专门使用pytorch的虚拟环境)

理解:在anaconda的大楼里建立一个使用pytorch工具的新房间,base可理解为anaconda大楼的大厅,我们可以通过搭建不同的房间,来安装不同包,执行不同的项目。

 步骤:

 step1. 打开Anaconda的命令行界面控制端Anaconda Promp,输入以下指令,建立新环境

conda create -n 【name_of_new_environment】python=【python_version】

pytorch的深度学习环境安装配置(GPU版)

新的虚拟环境pytorch的地址:

pytorch的深度学习环境安装配置(GPU版)

 step2. 输入”y“,在新环境中下载以上建议的new packages.

pytorch的深度学习环境安装配置(GPU版)

安装成功后,Anaconda的envs目录下就多了一个名为pytorch的文件夹。

pytorch的深度学习环境安装配置(GPU版)

 其他指令:

 (1)进入新环境

conda activate 【虚拟环境名】

(2) 查看该环境下装的包

conda list

(3) 删除创建的虚拟环境

conda remove -n 【虚拟环境名字】 --all

 2. 更新显卡驱动CUDA Driver

pytorch的深度学习环境安装配置(GPU版)

(图片from B站:我是土堆) 

step1. 确定自己显卡的型号(算力)

        任务管理器中查看NVIDIA显卡型号,我的电脑是NVIDIA GeForce MX150

pytorch的深度学习环境安装配置(GPU版)

step2. 安装/更新驱动CUDA Driver 的最新版本

(电脑有显卡驱动的话也可以跳过这一步,但是还是建议更新,比如我电脑自带的驱动版本就很低(391.25),导致我的驱动支持的最高CUDA Runtime版本(9.1)也很低,进而导致我能安装pytorch版本就很低)

打开NVIDIA官网,根据自己的显卡型号下载官方驱动:
https://www.nvidia.cn/Download/index.aspx?lang=cn

pytorch的深度学习环境安装配置(GPU版)

安装一路默认next就行了。

step3. 查看自己的驱动CUDA Driver版本,即支持的最高CUDA Runtime版本(我这里是12.0)

法(一):Dos窗口输入以下指令,查看显卡驱动版本

nvidia-smi

pytorch的深度学习环境安装配置(GPU版)

 法(二):右击打开NVIDIA控制面板->查看(左下角)系统信息

pytorch的深度学习环境安装配置(GPU版)

step4. 打开Pytorch官网,根据自己电脑支持的最高CUDA Runtime版本,选择对应pytorch版本(比CUDA driver小就行,确保CUDA Driver版本>=CUDA Runtime版本)

 3. 安装pytorch

PyTorch官网https://pytorch.org/当前支持的稳定最新版本是1.13.1,支持CUDA11.6、CUDA11.7(<12.0)。 

 包括4个包:
pytorch:核心包,集成pytorch的核心功能

torchvision:提供给图像处理的包

torchaudio:提供给语言处理的包

pytorch-cuda=11.6:CUDA的一些工具

3.1 法(一):利用pip安装Pytorch

pytorch的深度学习环境安装配置(GPU版)

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

命令解析:使用地址 "https://download.pytorch.org/whl/cu116" 安装以下 3个包.

  •         torch:核心包,集成pytorch的核心功能
  •         torchvision:提供给图像处理的包
  •         torchaudio:提供给语言处理的包

3.1.1 法(一)在线pip安装

step1. 进入新建的pytorch虚拟环境

pytorch的深度学习环境安装配置(GPU版)

 step2. 直接复制官网的pip指令至命令行窗口下载

pytorch的深度学习环境安装配置(GPU版)

(速度还是挺快的。)

pytorch的深度学习环境安装配置(GPU版)

 3.1.2 法(二)本地pip安装

step1. 从地址"https://download.pytorch.org/whl/cu116"下载(迅雷下载快)torch package

pytorch的深度学习环境安装配置(GPU版)

step2.使用pip install本地安装torch package

pip install 【torch package的下载路径地址】

step3. 本地安装完成后,再复制官网的pip指令至命令行窗口,安装剩下的package ,包小,下载会很快。

(它会自动检测,跳过安装好的torch包)

3.2 法(一):利用conda安装Pytorch

pytorch的深度学习环境安装配置(GPU版)

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

命令解析:使用pytorch通道下载安装以下 4个包(-c nvidia是什么意思?nvidia通道?)

  •         pytorch:核心包,集成pytorch的核心功能
  •         torchvision:提供给图像处理的包
  •         torchaudio:提供给语言处理的包
  •         pytorch-cuda=11.6:cuda的工具

复制官网conda下载命令,采用命令行下载(好慢啊,可以添加镜像地址)

3.3 验证pytorch是否安装成功

step1. 在我们新建的pytorch虚拟环境下,输入以下命令,查看有没有对应package

conda list

pytorch的深度学习环境安装配置(GPU版)

step2. 输入python,进入python解释器

step3. 输入import torch,调用torch包

step4. 输入以下命令,查看是否可以使用GPU

torch.cuda.is_available()

pytorch的深度学习环境安装配置(GPU版)

 4. Pycharm配置虚拟环境

4.1 给新建项目配置虚拟环境

 新建项目-->配置项目的路径-->配置项目的python解释器-->找到conda的虚拟环境(前面新建的pytorch虚拟环境)下的python解释器。

PS:anaconda的base虚拟环境的python解释器在anaconda安装目录的根目录下;新建的虚拟环境在anaconda安装目录下的envs目录下。

pytorch的深度学习环境安装配置(GPU版)

pytorch的深度学习环境安装配置(GPU版)

 验证是否可以使用GPUpytorch的深度学习环境安装配置(GPU版)

4.2 给下载项目配置虚拟环境 

 导入下载的项目-->File-->Setings-->Project:-->Python Interpreter-->选择项目需要的虚拟环境

pytorch的深度学习环境安装配置(GPU版)

 参考:

https://www.bilibili.com/video/BV1S5411X7FY?p=1

        在各个教程博客中闲逛时,发现这个教程,神奇的B站,厉害的up主,讲得太详细了吧!他怎么知道我有哪些困惑的,谢谢谢谢!以上是我的学习理解记录笔记了。文章来源地址https://www.toymoban.com/news/detail-413901.html

到了这里,关于pytorch的深度学习环境安装配置(GPU版)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包赞助服务器费用

相关文章

  • (纯小白向)Windows配置GPU深度学习环境:Cuda+Anaconda+pytorch+Vscode

    (纯小白向)Windows配置GPU深度学习环境:Cuda+Anaconda+pytorch+Vscode

    目录 一、Cuda和Cudnn下载安装 1.1 确定自己的电脑显卡驱动支持的Cuda版本 1.2 Cuda下载与安装 1.3 Cudnn下载与安装 二、Anaconda下载安装 2.1 下载 2.2 安装 2.3 手动配置环境变量 2.4 测试是否安装成功 三、Pytorch下载安装 3.1 创建conda虚拟环境 3.2 Pytorch下载 四、Vscode下载与环境配置 4.1

    2024年02月05日
    浏览(57)
  • Anaconda配置深度学习环境并安装GPU版torch

    Anaconda配置深度学习环境并安装GPU版torch

    本人属于刚入学的小白,因为任务需要,所以得从零开始安装深度学习环境。对于从未接触过深度学习的人来讲,光配置环境就花费了我好久好久的时间,中间心态炸裂好几次,索性还是安装成功了。现在就从0开始复盘一下我的安装过程。不喜勿喷,出门右转不送。爷又不靠

    2024年02月06日
    浏览(12)
  • 【深度学习环境搭建】Windows搭建Anaconda3、已经Pytorch的GPU版本

    【深度学习环境搭建】Windows搭建Anaconda3、已经Pytorch的GPU版本

    无脑下载安装包安装(自行百度) 注意点: 1、用户目录下的.condarc需要配置(自定义环境的地址(别忘了给文件夹加权限);镜像源) 1、先看你的显卡版本 Win +R - 输入命令nvidia-smi,看你的cuda版本 2.下载离线版本安装包(在线也行,只要你有耐心) https://download.pytorch.org

    2024年02月02日
    浏览(37)
  • 深度学习GPU服务器环境配置

    组里面分配了台gpu服务器,让我负责管理,需要先配置一些深度学习环境,记录一下。 一些linux服务器常用命令:https://blog.csdn.net/qq_51570094/article/details/123949910 首先需要了解linux系统信息 通过 cat /proc/version 查看包含gcc的版本信息 通过 hostnamectl 命令查看系统信息,这个命令是

    2024年02月01日
    浏览(12)
  • windows11+GPU1060安装强化学习环境之pytorch

    这里大家看一个视频系列,讲得非常详细,链接在此:https://www.bilibili.com/video/BV1S5411X7FY?p=28vd_source=3be739b673e1151850f4b8060ac78e1a 这里主要是说我遇到的问题以及解决办法。 首先,我的笔记本是有显卡的,只不过算力为6,较低,但是已经差不多了,主要是想在windows上操作,嗯,还

    2024年01月19日
    浏览(9)
  • win10下wsl2使用记录(系统迁移到D盘、配置国内源、安装conda环境、配置pip源、安装pytorch-gpu环境、安装paddle-gpu环境)

    win10下wsl2使用记录(系统迁移到D盘、配置国内源、安装conda环境、配置pip源、安装pytorch-gpu环境、安装paddle-gpu环境)

    安装好后环境测试效果如下,支持命令nvidia-smi,不支持命令nvcc,usr/local目录下没有cuda文件夹。 系统迁移到非C盘 wsl安装的系统默认在c盘,为节省c盘空间进行迁移。 1、输出 wsl -l 查看要迁移的系统名称 2、执行导出命令: wsl --export Ubuntu-20.04 ./Ubuntu-20.04.tar ,以下命令将系统

    2024年02月20日
    浏览(9)
  • 跟着李沐学AI(动手学深度学习 PyTorch版)学习笔记——03安装(环境配置d2l、pytorch)(python3.7版本+Windows+各种问题解决措施)

    跟着李沐学AI(动手学深度学习 PyTorch版)学习笔记——03安装(环境配置d2l、pytorch)(python3.7版本+Windows+各种问题解决措施)

    1.下载Miniconda下载地址 2.在安装过程中需要勾选“Add Anaconda to the system PATH environment variable”选项 3.检验win+R,输入cmd,在文本框输入conda --version 1.点击该链接+点击jupyter记事本下载压缩包 2.解压该压缩包 3.在解压后的文件夹地址栏输入cmd回车进入命令模式。 1.conda和pip默认使⽤

    2024年02月12日
    浏览(15)
  • Python&aconda系列:GPU深度学习环境搭建:Win11+CUDA 11.7+Pytorch1.12.1+Anaconda以及对应版本

    Python&aconda系列:GPU深度学习环境搭建:Win11+CUDA 11.7+Pytorch1.12.1+Anaconda以及对应版本

    官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。 以下是Pytorch和CUDA对应的版本 CUDA 环境 PyTorch 版本 9.2 0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.0 1.2.0、1.1.0、1.0.0(1) 10.1 1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.2 1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.9.0、1.10.0、1.

    2024年02月02日
    浏览(43)
  • GPU深度学习环境搭建:Win10+CUDA 11.7+Pytorch1.13.1+Anaconda3+python3.10.9

    GPU深度学习环境搭建:Win10+CUDA 11.7+Pytorch1.13.1+Anaconda3+python3.10.9

    在命令行中输入【nvidia-smi】可以当前显卡驱动版本和cuda版本。 根据显示,显卡驱动版本为:Driver Version: 516.59,CUDA 的版本为:CUDA  Version 11.7。 此处我们可以根据下面的表1 显卡驱动和cuda版本对应关系得知,Driver Version: 527.47  516.31,所以可以使用 CUDA Version 11.7 CUDA工具版本

    2024年04月26日
    浏览(37)
  • 深度学习-环境搭建(安装Pytorch)

    深度学习-环境搭建(安装Pytorch)

    入门深度学习过程中,我决定在笔记本上搭建深度学习环境。我笔记本的显卡是NVIDIA 1050Ti,似乎足以支撑。由于我之前已经安装过Anaconda,所以是要在Anaconda上安装GPU版本的pytorch 略。 大多数人更推荐安装的是Miniconda,特别适合新手,能够节省空间和时间。 右键打开NVIDIA控制

    2023年04月26日
    浏览(13)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包