
GPU
JasonLiu1919
微信公众号《小窗幽记机器学习》。却顾所来径,苍苍横翠微。个人微信号:onepieceand
展开
-
docker拉取的pytorch-gpu版找不到cuda和cudnn的位置,为何?
问题描述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名不同的版原创 2020-05-19 11:01:46 · 9945 阅读 · 6 评论 -
nvidia-smi 和 nvcc 结果的版本为何不一致
背景为何nvidia-smi 中的CUDA 版本与 nvcc不一致:从上述结果可以看出,nvidia-smi的结果显示CUDA版本是10.0,而从nvcc命令来看,却是CUDA 9.0分析其实是因为CUDA 有两种API,分别是 运行时 API 和 驱动API,即所谓的 Runtime API 与 Driver API。nvidia-smi 的结果出了 GPU 驱动版本型号,还有 CU...原创 2019-10-19 17:19:35 · 37029 阅读 · 11 评论