Win11+Docker搭建CUDA开发环境

2023-12-24 19:50:32

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

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

本人软硬件条件

  • OS: win11
  • GPU: RTX 3060
  • Driver Version: 537.42
  • CUDA Version: 12.2
  • Docker: Dokcer Desktop 4.12.0

目前想在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


参考

文章来源:https://blog.csdn.net/transformer_WSZ/article/details/135184754
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。