search

shenuiuin / LXD_GPU_SERVER

实验室GPU服务器的LXD虚拟化
401 stars 65 forks source link

想请问一下，容器内pytorch调用cuda总是提示初始化失败，而且torch.cuda.is_available总是false #18

Closed Pradur241 closed 2 years ago

Pradur241 commented 2 years ago

image image image

Pradur241 commented 2 years ago

cuda和驱动都正常，是因为宿主机没装cuda的问题嘛？

Pradur241 commented 2 years ago

解决了，重新安装gcc和g++，重装驱动就能通过cuda的编译，torch也能识别cuda