pytorch学习系列——环境搭建(代码片段)

非晚非晚 非晚非晚     2023-01-07     193

关键词:


由于使用的需要,从现在开始陆续学习并记录pytorch相关的使用,首先这篇介绍pytorch的环境搭建,全部内容都经过实际操作,而且都是以最简单、安全、高效的方式进行安装。

1. 安装英伟达驱动

  • 查看笔记本推荐的驱动版本

使用下列命令进行查看

sudo ubuntu-drivers devices

可以看到我的电脑推荐的是nvidia-driver-470

  • 安装

可以直接使用ubuntu自带的软件和更新进行安装的,如下图所示:

2. 安装Anaconda环境

  • 命令行安装
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sh Miniconda3-latest-Linux-x86_64.sh

这种安装方式会直接从官网下载安装,可能会比较慢。如果想要速度快一点,还可以选择镜像下载脚本安装,清华镜像地址

  • 环境变量
sudo gedit /etc/profile

行尾添加:

export PATH=/home/liqiang/miniconda3/bin:$PATH
  • 重启,并验证是否安装成功
conda –V

3. 安装pytorch、CUDA和cuDNN环境

(1)配置国内镜像加速

如果直接执行官网给定的安装命令,通常情况下,会很慢,可以使用国内镜像加速。

首先打开文件:

sudo gedit ~/.condarc

然后将里面的全部内容替换为:

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
show_channel_urls: true

(2)安装

pytorch安装还是挺简单的,直接去官网,选择相应的版本,就可以安装了。

下图是最新版的pytorch,也可以选择以前的版本进行安装。

安装方法如上图,官网直接下载命令如下(下列命令会默认从官网下载,不管是否设置了国内镜像):

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

如果使用的国内镜像,请使用下列命令

conda install pytorch torchvision torchaudio cudatoolkit=11.1

然后会出现下列界面,等待安装完成即可。

4. 查看安装环境的版本

使用上述安装过程默认会安装CUDA和cuDNN。下面检查它们是否被安装成功。

4.1查看Anaconda版本

conda -V

会出现下列界面:

4.2 查看Nvidia 驱动版本

nvidia-smi

会出现下列界面

4.3 查看pytorch版本

打开终端,并输入python,进入python交互式环境,然后输入下列命令:

import torch
print(torch.__version__)  #注意是双下划线

操作如下:

4.4 查看CUDA版本

如果是手动安装的话,可以输入下列命令查看cuda版本

cat  /usr/local/cuda/version.txt

如果是使用conda,也就是本文介绍的方法安装,可以使用pytorch程序进行查看。

首先进入python交互式界面,然是输入下列命令:

import torch
print(torch.version.cuda)

操作如下:

4.5 查看cuDNN版本

和CUDA一样,如果是手动安装则使用下列命令即可查看:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果是使用conda方法安装,可以使用python程序进行查看:

import torch
print(torch.backends.cudnn.version())

操作如下:

李沐《动手学深度学习》第二版pytorch笔记1环境搭建(代码片段)

李沐《动手学深度学习》第二版pytorch笔记1搭建环境文章目录李沐《动手学深度学习》第二版pytorch笔记1搭建环境此时尚有耐心虚拟环境搭建创建虚拟环境查看已有的虚拟环境激活虚拟环境安装深度学习框架和d2l软件包安装d2l安... 查看详情

深度学习环境搭建anaconda+pycharm+pytorch(代码片段)

文章目录显卡驱动cudaanaconda1.下载安装2.安装pytorch虚拟环境3.conda常用指令pycahrm/jupyter下载安装如何建好的虚拟环境的解释器找出来指派给代码?本文将详细介绍一下如何搭建深度学习所需要的实验环境.这个框架分为以下六个... 查看详情

深度学习-pytorch环境搭建(windows)(代码片段)

文章目录版本选择Python安装Anaconda安装PyTorch安装PyCham测试前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站。本文主要参考PyTorch文档版本选择不同操作系统&#... 查看详情

学习经验分享no.22pytorch环境搭建(代码片段)

    YOLO系列算法检测需要搭建PyTorch环境才能运行项目,本博客将手把手教学如何搭建PyTorch环境,以YOLOv5检测算法为例,进行讲解。有问题请关注后私信留言,可远程教学。一、 安装Anaconda、Cuda、pycharm1.安装Ana... 查看详情

三维深度学习pytorch-pointnet系列之win10下环境安装与demo运行(代码片段)

【三维深度学习】Pytorch-PointNet系列之win10下环境安装与demo运行提示:最近开始在三维深度学习方面进行研究,从PointNet开始入手,对这个系列的网络进入深入学习,记录相关知识点,分享学习中遇到的问题已经解决的方法。文章目录【... 查看详情

深度学习pytorch环境搭建(代码片段)

搭建pytorch环境,主要目的还是尽量利用gpu的算力,cpu也可以但速度要慢很多,如果没有Nvidia显卡,torch.cuda.is_available()就是False,是正确的。即便没有显卡,也是可以往后面学习的,只不过运算速度会慢... 查看详情

opencv学习系列(零)mac下opencv+xcode环境搭建(代码片段)

#OpenCV学习系列(零)Mac下OpenCV+xcode环境搭建[-=博客目录=-]1-学习目标1.1-本章介绍1.2-实践内容1.3-相关说明2-学习过程2.1-homebrew安装2.2-使用homebrew安装CMake2.3-源码安装OpenCV2.4-xcode配置与测试2.5-基础问题回答3-资料1-学习目标1.1-本... 查看详情

vue学习系列--环境搭建踩坑总结(代码片段)

今天在线下搭建开发环境时遇到了不少问题,废话少说,直接上截图并给出具体的解决方案。按照build.sh的指示,只需依次执行以下命令即可:#根据package.json文件里配置的依赖项,下载相应文件npminstall#重新编... 查看详情

pytorch学习笔记:pytorch进阶训练技巧(代码片段)

PyTorch实战:PyTorch进阶训练技巧往期学习资料推荐:1.Pytorch实战笔记_GoAI的博客-CSDN博客2.Pytorch入门教程_GoAI的博客-CSDN博客本系列目录:PyTorch学习笔记(一):PyTorch环境安装PyTorch学习笔记(二)... 查看详情

利用anaconda安装pytorch搭建深度学习环境(代码片段)

一、NVIDIA驱动安装 首先查看电脑的显卡版本,步骤为:此电脑右击-->管理-->设备管理器-->显示适配器。就可以看到电脑显卡的版本了。得知以上的信息以后我们就可以对应我们的显卡去英伟达官网上去找相对应的... 查看详情

pytorch学习笔记:模型定义修改保存(代码片段)

往期学习资料推荐:1.Pytorch实战笔记_GoAI的博客-CSDN博客2.Pytorch入门教程_GoAI的博客-CSDN博客本系列目录:PyTorch学习笔记(一):PyTorch环境安装PyTorch学习笔记(二):简介与基础知识PyTorch学习笔记&#... 查看详情

pytorch-入门windows下的环境搭建(经验证成功~)(代码片段)

...的服务器都是windows环境TT...】,但身边的师兄们用的都是pytorch,自己查了查现在做科研基本上都是用的pytorch,而且现在pytorch的windows版本也已经很成熟了,fastai深度学习库也受到了广泛的好评,所以...果断转!环境搭建-windows-gp... 查看详情

搭建深度学习环境(pytorch)

...习环境。(2).按照Anaconda3+CUDA10.0+CuDNN+Pytorch1.2+Pycharm配置。(3).解决配置Pytorch中无法使用torchvis 查看详情

搭建深度学习环境(pytorch)

...习环境。(2).按照Anaconda3+CUDA10.0+CuDNN+Pytorch1.2+Pycharm配置。(3).解决配置Pytorch中无法使用torchvis 查看详情

pytorch学习系列——加载数据2(代码片段)

...数据MyDataset,那么怎么在训练时提供batch数据呢?PyTorch提供了生成batch数据的类。PyTorch用类torch.utils.data.DataLoader加载数据,并对数据进行采样,生成batch迭代器。classtorch.utils.data.DataLoader(da 查看详情

anaconda配置各版本pytorch深度学习环境(代码片段)

目录1.前言2.配置镜像源3.pytorch,torchvision,python版本对应4.创建并进入虚拟环境5.Pytorch0.4.16.Pytorch1.0.07.Pytorch1.0.18.Pytorch1.1.09.Pytorch1.2.010.Pytorch1.4.011.Pytorch1.5.012.Pytorch1.5.113.Pytorch1.6.0 查看详情

在ubuntu下将pytorch模型部署到c++(环境搭建)(代码片段)

...装opencv安装及编译本文总结记录了在将lightweight-openpose的pytorch网络模型部署到c++的过程以及遇到的一些问题,如有错误敬请指正vscodec++环境部署cmakevscode中c++编译过程主 查看详情

深度学习系列50:苹果m1芯片加速pytorch(代码片段)

1.介绍Apple的MetalPerformanceShaders(MPS)作为PyTorch的后端来加速GPU训练。MPS后端扩展了PyTorch框架,提供了在Mac上设置和运行操作的脚本和功能。MPS通过针对每个MetalGPU系列的独特特性进行微调的内核来优化计算性能。新... 查看详情