服务器上通过podman容器运行需要较低版本cuda的代码
在本地跑代码会爆显存,所以想用组里的服务器来训练模型,由于使用的事tensorflow的gpu版本,所以需要对应的cuda和cudnn版本。使用的是Red hat的服务器,由于服务器的cuda版本较新,所以无法运行tensorflow-gpu1.2.0支持的代码,而且我没有root权限,所以也无法直接安装自己所需的cuda和cudnn版本。于是就想着借助podman容器来运行我的点代码。
原创
2024-08-22 10:35:20 ·
488 阅读 ·
0 评论