搭建PyTorch炼丹环境（踩坑记录）

最新推荐文章于 2024-05-02 15:44:37 发布
最新推荐文章于 2024-05-02 15:44:37 发布
阅读量1.1k 收藏 2
点赞数
文章标签： docker pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/qq_27075947/article/details/100715284
版权

目录

1. 使用docker搭建PyTorch运行环境

可以在docker hub上选择合适的镜像，注意选择的镜像CUDA版本尽量与服务器宿主机一致，例如：在RTX 2080 Ti上，使用CUDA 9的版本（比如：floydhub/pytorch:1.0.0-gpu.cuda9cudnn7-py3.39），就会报以下错：

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

其实，不使用GPU，只用CPU的话，CuDnn版本不一致，代码也可以运行。
所以，为了GPU加速，我们还是选择CUDA 10的docker镜像

docker pull pytorch/pytorch:1.0.1-cuda10.0-cudnn7-devel

使用nvidia-docker来run一个容器，所以还需自备nvidia-docker。

2. 直接在服务器上安装PyTorch

Yumath
关注
C#深度学习革命：用PyTorch点燃你的AI代码之魂
墨夶的博客
04-16 276
使用PyTorch的C#接口实现MNIST手写数字分类。创建并操作张量，理解PyTorch的“灵魂”用PyTorch在C#中实现线性回归模型。模型训练速度慢，CPU负载过高。在C#中使用GPU加速模型训练。
深度学习纯小白如何从零开始写第一篇论文？看完这篇豁然开朗！_计算机深度学习如何写出一篇论文(1)
2401_84248681的博客
05-02 1156
论文摘要就是整篇文章和浓缩预览，它被排放在论文的首要位置，也是审稿人首先看的地方。公式：研究的目的和重要性**+研究的方法+得到的研究结果****+结论**⚡️摘要是一整段不要在摘要里出现感情色彩的评价一般在200-300字左右，具体看期刊。
PyTorch安装 -Python深度学习模块-各种问题
一个爱学习的小小白
03-27 1972
打开网址https://pytorch.org/get-started/locally/ 找到对应的版本 由于我的CUDA是10.1版本 所以我应该选择“none”(选择对应的cuda很重要) 由于Anoconda安装网络下载经常会出现各种各样的失误 所以我们都选择用pip 安装。 加粗样式 打开王也 打开这个网页下载whl文件，放到 这个文件夹里，然后 重要的步骤来了 1、打开Anoconda...
炼丹环境搭建知识汇总
大龙的编程学习笔记
06-02 187
@[TOC]自动根据文章标题生成目录 tf 使用conda，一条命令即可安装tf-gpu版本 conda install tensorflow-gpu https://mp.weixin.qq.com/s?__biz=MzAxMzUzOTg3OQ==&mid=2455934821&idx=1&sn=616f6795a18d78b429fb9fd76876cc93 pytorch 使用conda，一条命令即可安装pytorch，在pytorch官网上可以查找相关命令 https://w
pytorch 命令行运行_pytorch: 使用docker
weixin_39637370的博客
02-04 626
为了不用搭环境去使用GPU服务，然鹅太费钱了，每一分钟都在烧钱，扛不住了，决定自己搭环境吧，毕竟有GPU可用。1. 服务器上已有的环境就是docker集群，所以很简单，pull一个pytorch的镜像就可以开始啦去官网找合适的pytorch镜像版本，服务器上cuda是11.0的，但官网上面没有该版本，我选择了10.1版本的。docker pull pytorch/pytorch:1.3-cuda1...
PyTorch 安装 GPU版本(CUDA/cuDNN)
あずにゃん梓喵的博客
07-30 964
日萌社 人工智能AI：Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战（不定时更新） 1.官网下载安装：https://pytorch.org/get-started/locally/ 2.执行命令：conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 3.比如我...
语音、机器学习环境搭建记录
Haulyn5的博客
09-09 1469
上一个 docker 容器让我安装奇奇怪怪的库的时候，动了一个系统库的文件，然后之间用的 ohmyzsh 给崩了……正好上一个容器忘了多开几个端口，这次重新搭建一个环境，把零零碎碎的东西记录下来。 主要安装了 miniconda, pytorch, coqui tts
深度学习纯小白如何从零开始写第一篇论文？看完这篇豁然开朗！_计算机深度学习如何写出一篇论文
最新发布
2401_84248681的博客
05-02 1376
论文摘要就是整篇文章和浓缩预览，它被排放在论文的首要位置，也是审稿人首先看的地方。公式：研究的目的和重要性**+研究的方法+得到的研究结果****+结论**⚡️摘要是一整段不要在摘要里出现感情色彩的评价一般在200-300字左右，具体看期刊。
深度学习之YOLOv5实践应用（1）系统安装配置
light169的专栏
03-09 3895
【动手学深度学习（一）】Yolov5环境配置教程 - 知乎（ubuntu） [译] 再看 Flask 视频流 - 知乎 使用YOLOv5实现多路摄像头实时目标检测 yolov5检测视频流的原理、detect.py解读 yolov5检测视频流的原理、detect.py解读 vue+flask实现视频目标检测yolov5 Flask部署YOLOv5 - 知乎 【Yolov5】训练自己的yolov5模型并集成到安卓应用中【上】在WEB端部署YOLOv5目标检测（Flask+VUE） 实用教程详...
超坑！安装CUDA与pytorch版本匹配的问题
热门推荐
YuanYuan Code
04-20 7万+
慢悠悠地装了很久，觉得CUDA和pytorch官网标注的当前该pytorch所匹配的CUDA应该是完全一致的，所以一开始就安装 10.1.105版本的CUDA，pytorch为1.4（官方说适配CUDA版本为10.1嘛~）： 但是事实证明，不行。 最后还是降级安装pytorch，从1.4（对应CUDA10.1）到1.2版本的pytorch（对应CUDA10.0），还是不行，最后： 显卡驱动版本号...
深度学习Pytorch(一)
stormxie的博客
03-08 184
此文章为我学习深度学习Pytorch时所做的笔记记录
最全tensorflow，PyTorch ，numpy和keras 版本匹配汇总
源代码杀手的博客
08-06 2万+
Framework Env name (--env parameter) Description Docker Image Packages and Nvidia Settings TensorFlow 2.2 tensorflow-2.2 TensorFlow 2.2.0 + Keras 2.3.1 on Python 3.7. floydhub/tensorflow TensorFlow-2.2 TensorFlow 2.1 tensorflow-2.1 TensorFlow 2.1.0 + Kera.
docker pytorch-cuda 安装 devel 版本
qq_28424679的博客
04-23 4059
查看镜像 docker image ls docker 镜像文档查询链接 Docker Hubhttps://hub.docker.com/r/pytorch/pytorch/tags 选择自己合适的版本进行安装（直接复制即可获取安装代码） 拉镜像： docker pull pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel 查看镜像 docker image ls 重新命名： docker tag 7554ac65eba5 jianhu.
docker拉取的pytorch-gpu版找不到cuda和cudnn的位置，为何？
ljp1919的专栏
05-19 1万+
问题描述 pytorch 镜像位置： https://hub.docker.com/r/pytorch/pytorch/tags 拉取镜像： docker pull pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime 查看本地现有镜像清单： 创建一个容器： docker run --gpus all -td --name=liujiepeng_pytorch pytorch/pytorch bash 注意：其实这里有两个repository名字相同，tag名不同的版
Docker 在Tesla K80 和 RTX 2080配置CUDA，Pytorch过程，cuda runtime error (11)解决方案
weixin_40774341的博客
05-16 6045
Docker 在Tesla K80 和 RTX 2080配置CUDA，Pytorch过程Tesla K80RTX 2080 第一次使用docker，需要跑一个深度学习程序，要求pytorch==1.0.1，因此开始在两台服务器不同显卡上配置docker Tesla K80 先pull镜像，镜像在https://github.com/anibali/docker-pytorch有说明，直接执行以下命...
floydhub使用手册
Songerggggggg的博客
01-14 383
floydhub提供算力，支持开发者上传深度学习代码以及数据集，训练网络，具体价格如下： 本文记录自己训练网络所使用到的一些常规基本操作，作为备忘录。 首先win+r 打开cmd命令窗口， 安装floyd，具体命令为： pip install -U floyd-cli 安装成功后，直接登录，输入： floyd login 此时会跳转出网页， 我们继续在cmd里面操作，在项目所在文件夹下面的cmd中输入： floyd init pls （注：pls是我们项目的名称） 之后我
docker 镜像启动成功但是无法访问
zhaxiaodong的博客
01-16 3万+
docker 镜像启动成功但是无法访问 解决办法： vi /etc/sysctl.conf 或者 vi /usr/lib/sysctl.d/00-system.conf 添加如下代码：     net.ipv4.ip_forward=1 重启network服务 systemctl restart network 查看是否修改成功 sysctl net.ipv4.ip_forwa...
docker挂载NVIDIA显卡运行pytorch
zhulianhai0927的专栏
04-16 2715
来自：https://www.cnblogs.com/mar-q/p/8417184.html 写在前面： 　　请参考之前的文章安装好CentOS、NVIDIA相关驱动及软件、docker及加速镜像。 　　主机运行环境 $ uname -a Linux CentOS 3.10.0-514.26.2.el7.x86_64 #1 SMP Tue Jul 4 15:04:05 UTC 20...
分步指南：搭建PyTorch环境与CUDA版本检查
"本文主要介绍了如何在计算机上搭建PyTorch环境，包括使用conda创建虚拟环境、切换到PyTorch环境、安装常用库以及下载和验证PyTorch和CUDA的安装。" 在进行深度学习开发时，PyTorch是一个常用的框架，而良好的环境...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值