基于 cuda 开发是咋么控制工具链版本的。

167 天前
 SkywalkerJi
nv 的依赖也太长了,驱动 cuda cuDNN tensor numpy torch ,可能一个小版本号变化就直接不向前兼容。
conda 只能控制住 python 包和 pip 依赖的版本,cuda 那些直接装在 Windows 系统里的环境有什么好办法控制吗?
1461 次点击
所在节点    Python
7 条回复
brucedone
167 天前
基于 docker 镜像的环境控制
zhusimaji
167 天前
目前是 docker 镜像,但是也得看宿主机驱动啥的更新到多少,低的也用不了高版本吧
chapiom
167 天前
这些用 conda 已经可以装了,类似这样的
conda install conda-forge::cuda-runtime=12.4.1 cudnn=8.9.2.26 conda-forge::gputil=1.4.0
SkywalkerJi
167 天前
@brucedone #1
@zhusimaji #2
docker 里面跑,性能不会受影响吗?另外我还需要用到一些 Windows sdk ,docker 里也可以么?
zhusimaji
167 天前
@SkywalkerJi 啊这,用 win sdk 啊,我都是在 linux 下跑模型训练脚本,win sdk 超出认知了
brucedone
167 天前
@SkywalkerJi 如果你是 windows container ,理论上不受影响,但 win 的 container 不知道是否支持 cuda
clemente
159 天前
nv-docker 他们已经打包好了 开箱就用

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/1055602

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX