问题描述

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名不同的版本。可以通过用repository:tag进行区分。

进入容器内部：

docker exec -it liujiepeng_pytorch bash

发现在内部找不到cuda和cudnn的位置？

可以看出，有CUDA的驱动(10.2)，但是找不到lib文件。也没有nvcc。但是在pytorch内部查看cuda版本：

可以看出cuda版本是10.1，这与拉取的镜像版本一致。

原因分析

究其原因是拉取的镜像是runtime版，所以只有runtime版本的cuda，并没有compiler版。这与安装带有CUDA libs的Pytorch二进制文件类似。Pytorch可以在GPU上运行，但是此时不需要安装完整的CUDA libs，所以也就不能在这个运行环境之外使用nvcc。

解决方案

如果拉取dev版：

docker pull pytorch/pytorch:1.5-cuda10.1-cudnn7-devel

则可以如下图看到lib文件的位置：

或者直接自己在容器中安装cuda

docker拉取的pytorch-gpu版找不到cuda和cudnn的位置相关推荐

Docker 拉取镜像慢解决
问题今天使用 docker 拉取 mysql:5.7 发现很慢并且卡住了,无奈之下找解决办法,后来找到可以用阿里云容器镜像服务里的镜像加速器解决解决访问阿里云容器镜像服务网站按照操作文档执行 ...
Docker 拉取镜像及标签 pull | tag
Docker 拉取镜像及标签 pull | tag 重翻Fabric项目的源码,发现Docker部分内容,有很多不尽理解的地方,看着看着,就看到使用docker pull拉取Fabric镜像及使用do ...
配置了docker镜像加速，但是docker拉取镜像速度还是很慢
首先影响docker拉取镜像加速的原因有几个,挨个进行检查后就会发现问题所在 1.添加docker的阿里yum源 tee /etc/yum.repos.d/docker.repo <<-' ...
深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】
深度学习环境配置入门深度学习,首先要做的事情就是要搭建深度学习的环境.不管你是Windows用户,Mac用户还是Ubuntu用户,只要电脑配置允许,都可以做深度学习,毕竟Windows.Mac和Ub ...
[Docker]Docker拉取,上传镜像到Harbor仓库
需求因为项目的需求,需要制作一个基于tomcat的镜像.那么前提就是,需要有tomcat的基础镜像. 怎么做我的思路跑偏了,本来以为是需要将tomcat下载下来,然后通过docker命令,让它成为 ...
docker 拉取oracle_Docker 拉取 oracle 11g镜像配置-Go语言中文社区
话不多说开始记录docker拉取阿里的oracle11g 镜像并进行配置, 用pl/sql 可以登录为最终结果 navicat连接是在最后一步但是根据这个进行配置会有一些问题,所以写这篇记录一下, ...
Docker拉取RabbitMQ镜像运行启动
Docker拉取rabbitmq镜像运行启动 RabbitMQ简介一.命令拉取二.获取镜像三.运行命令参数四.查看正在运行的容器五.客户端访问最后 RabbitMQ简介以熟悉的电商场景为 ...
Docker拉取镜像报错error pulling image configuration
Docker拉取镜像报错error pulling image configuration 在制作docker镜像的时候出现如下报错: xxx@ubuntu:~/Desktop/a$ docker b ...
docker拉取（下载）数据太慢，docker pull xxxx一直卡着。
docker pull xxxx,docker拉取(下载)数据太慢可能因为网络波动导致拉取失败或者一直卡着. 添加国内镜像,然后点击Apply&restart 即可! "regis ...

docker拉取的pytorch-gpu版找不到cuda和cudnn的位置

问题描述

原因分析

解决方案

docker拉取的pytorch-gpu版找不到cuda和cudnn的位置相关推荐

最新文章

热门文章