1.问题
想要将cuda9.0的docker部署到3090上,尝试多次重新安装pytorch,但是都出现问题:
RuntimeError: cuda runtime error (8) : invalid device function at /pytorch/torch/lib/THC/generic/THC
2.解决
重新编译高版本cuda的docker。可以使用的docker images:
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu16.04