Win11+Docker搭建CUDA开发环境

最新推荐文章于 2024-06-25 15:42:27 发布

transformer_WSZ

最新推荐文章于 2024-06-25 15:42:27 发布

阅读量1.4k

点赞数 8

分类专栏： LLM 文章标签： docker 容器运维

本文链接：https://blog.csdn.net/transformer_WSZ/article/details/135184754

版权

21 篇文章 0 订阅

订阅专栏

最近入门了CUDA编程，先记录下搭建环境过程。

由于在windows和wsl上折腾了好久，装cuda、cudnn、cmake、gcc等软件，还经常遇到依赖、版本许多问题，最终污染了系统环境。在朋友的安利下，采用docker容器开发方案，试一下真香。

目前想在docker容器里调用windows gpu，已经不再需要安装镜像nvidia-docker了。新版docker已经支持透传gpu，直接在参数里添加 --gpus all 即可：

docker run -it --gpus all --name gpu_test -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all mortals/codeenv:conda-cuda11.8

Dockerfile见：base-cuda11.8

关注

专栏目录