PyTorch环境配置

本文档用于记录PyTorch环境配置过程中的要点。

安装Miniconda

  • 下载Miniconda安装包:Miniconda3-latest-Linux-x86_64.sh
  • 修改安装包的文件权限属性,使得Miniconda安装包能够执行:chmod u+x Miniconda3-latest-Linux-x86_64.sh
  • 安装Miniconda: sh ./Miniconda3-latest-Linux-x86_64.sh

安装PyTorch

  • 通过Miniconda创建一个新的环境:conda create --name torch Python
  • 激活环境:conda activate torch
  • 查看CUDA版本:nvcc --version
  • 根据CUDA版本安装相应的PyTorch库:conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
  • 验证PyTorch是否安装成功:
    import torch
    x = torch.rand(5, 3)
    print(x) torch.cuda.is_available()

配置远程开发工具

本节用于记录配置VSCode远程编程与调试的插件Remote - SSH的过程。

  • 通过ssh-keygen命令在本地生成用于ssh公钥认证的公钥和私钥文件:ssh-keygen -t rsa -P '' -f ~/.ssh/id_rsa
  • 通过ssh-copy-id命令把公钥追加到远程机器的authorized_keys文件末尾: ssh-copy-id hello-kitty@192.168.***.***
  • 安装插件Remote - SSH
  • 配置插件:
    # C:\Users\****\.ssh\config
    Host ****@192.168.***.***
    HostName 192.168.***.***
    User ****

    这里需要注意配置项Host需要按照"用户@IP地址"的格式进行设定,否则远程连接时会出现“过程试图写入的管道不存在”的异常状况

基于CUDA的张量

  • 查看可用的GPU的数量:torch.cuda.device_count()
  • 定义CUDA device object,存在三种略有不同的方式:
    # 第一种方式
    device = torch.device('cuda:1') # 第二种方式,参数"1"表示显卡的下标(第二块显卡)
    torch.cuda.set_device(1)
    device = torch.device('cuda') # 第三种方式
    device = torch.device('cuda', 1)
  • 定义张量:x = torch.randn(5, 3, device=device)

导入警告问题

用VSCode编写Python代码,在导入语句import torch下方会出现黄色波浪线,提示未解决的导入“unresolved import 'torc'”。异常情况是由Microsoft Python Analysis Engine引发的,通过设置Jedi作为IntelliSense引擎,可以解决这一问题:

  • Ctrl + Shift + P,打开VSCode命令面板
  • 输入“Configure Language Specific Settings”,选择语言“Python”
  • 在打开的配置文件(settings.json)中找到配置项“python.jediEnabled”
  • 设定配置项"python.jediEnabled": true,,保存并重新加载VSCode

参考资料

  1. VScode Remote远程开发与调试
  2. Linux和Windows查看CUDA和cuDNN版本
  3. Linux chmod命令
  4. Start Locally | PyTorch
  5. vscode unresolved import warning

PyTorch环境配置的更多相关文章

  1. PyTorch环境配置及安装

    环境配置 温馨提示:为了更好的教程体验,提供视频.阅读地址 Youtube: https://www.youtube.com/playlist?list=PLgAyVnrNJ96CqYdjZ8v9Yj ...

  2. 计算机视觉2-> 深度学习 | anaconda+cuda+pytorch环境配置

    00 想说的 深度学习的环境我配置了两个阶段,暑假的时候在一个主攻视觉的实验室干活,闲暇时候就顺手想给自己的Ubuntu1804配置一个深度学习的环境.这会儿配到了anaconda+pytorch+c ...

  3. PyTorch深度学习入门笔记(一)PyTorch环境配置及安装

    @ 目录 一.工具安装 1.1 Anaconda 安装 1.2 Pytorch安装 二.编辑器安装 2.1 Pycharm安装 2.2 Jupyter安装 OS: ubuntu 20.04(虚拟机) ...

  4. pytorch 环境配置

    一.下载Anaconda 二.添加清华镜像 # 添加清华镜像 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anac ...

  5. lightweight openpose 入门实操笔记(pytorch环境)

    最近有个小项目要搞姿态识别,简单调研了一下2D的识别: 基本上是下面几种 (单人)single person 直接关键点回归 heatmap,感觉其实就是把一个点的标签弄成一个高斯分布 (多人)mul ...

  6. 超算云(GPU服务器)环境配置

    最近在用并行超算云GPU服务器(中国国家网格12区)搭建毕设的环境,这里记录一下. 首先,超算云服务器的登录可以采用网页版.也可以采用客户端(超算云地址:https://cloud.paratera. ...

  7. 【深度学习】PyTorch CUDA环境配置及安装

    Pytorch版本介绍 torch:1.6 CUDA:10.2 cuDNN:8.1.0 安装 NVIDIA 显卡驱动程序 一般 电脑出厂/装完系统 会自动安装显卡驱动 如果有 可直接进行下一步 下载链 ...

  8. 手把手教会你远程Linux虚拟机连接以及配置pytorch环境。

    出一期用于连接远程Ubuntu系统并配置pytorch环境的教学.2021-07-07 13:35:57- 现在的矿难导致显卡大幅度的涨价对很多要做深度学习领域的小伙伴们非常的不友好,配置设备固然要掏 ...

  9. Pytorch及Yolov5环境配置及踩坑

    Pytorch及Yolov5环境配置及踩坑 1.何为Yolov5 yolo是计算机视觉方面用来进行目标检测的一个十分出名的开源框架,我搜不到官方的对此概括性的定义,但实际上也没什么必要,更重要的是会使 ...

随机推荐

  1. Flash XSS

    示例: 创建一个测试flash,代码如下,导出swf文件 //取m参数 var m=_root.m; //取showInfo参数 var showInfo=_root.showInfo; //调用ht ...

  2. 【freertos】002-posix模拟器设计与cortex m3异常处理

    目录 前言 posix 标准接口层设计 模拟器的系统心跳 模拟器的task底层实质 模拟器的任务切换原理 cortex M3/M4异常处理 双堆栈指针 双操作模式 栈帧 EXC_RETURN 前言 如 ...

  3. 羽夏看Win系统内核——异常篇

    写在前面   此系列是本人一个字一个字码出来的,包括示例和实验截图.由于系统内核的复杂性,故可能有错误或者不全面的地方,如有错误,欢迎批评指正,本教程将会长期更新. 如有好的建议,欢迎反馈.码字不易, ...

  4. 在 Spring中如何注入一个java集合?

    Spring提供以下几种集合的配置元素: 类型用于注入一列值,允许有相同的值. 类型用于注入一组值,不允许有相同的值. 类型用于注入一组键值对,键和值都可以为任意类型. 类型用于注入一组键值对,键和值 ...

  5. jQuery--筛选【查找函数】

    查找函数介绍 <A> <B> <C></C> <D></D> <E></E> <F>< ...

  6. SpringDataJpa使用审计(Auditing)功能

    SpringBoot项目使用SpringDataJpa提供的审计功能的使用流程 SpringDataJpa提供审计注解:@CreatedBy,@LastModifiedBy,@CreatedDate, ...

  7. (原创)[C#] 一步一步自定义拖拽(Drag&Drop)时的鼠标效果:(一)基本原理及基本实现

    一.前言 拖拽(Drag&Drop),属于是极其常用的基础功能. 无论是在系统上.应用上.还是在网页上,拖拽随处可见.同时拖拽时的鼠标效果也很漂亮,像这样: 这样: 还有这样: 等等等等. 这 ...

  8. HTTP-完整状态码表

    HTTP状态码列表: 状态码 状态码英文名称 中文描述 100 Continue 继续.客户端应继续其请求 101 Switching Protocols 切换协议.服务器根据客户端的请求切换协议.只 ...

  9. Asp.Net Core之Identity应用(上篇)

    一.前言 在前面的篇章介绍中,简单介绍了IdentityServer4持久化存储机制相关配置和操作数据,实现了数据迁移,但是未对用户实现持久化操作说明.在总结中我们也提到了, 因为IdentitySe ...

  10. webpack+vue-cli+ElementUI+vue-resource 前端开发

    线上商城项目(电脑0环境)1.安装node.js (参考http://www.runoob.com/nodejs/nodejs-install-setup.html): 2.安装 webpack (全 ...