docker tensorflow-gpu cuda10.2踩坑

cuda10.2对应tensorflow-gpu版本2.1

tensorflow-gpu 2.1不支持python3.8

解决方法:

docker pull opensciencegrid/tensorflow-gpu:2.3-cuda-10.1

进入容器,更改默认python2.7为3.6

update-alternatives --install /usr/bin/python python /usr/bin/python3 150


版权声明:本文为weixin_38906715原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。