tensorflow与pytorch GPU版在同一环境中混合安装

在同一环境中安装tensorflow和pytorch的GPU版

找到各自的一个版本,使得能共用同一版本的cuda,下面是一个可用的版本组合

NVIDIA GPU驱动: Driver Version: 440.44
conda install tensorflow-gpu=1.15.0
conda install pytorch torchvision cudatoolkit=10.0

推荐阅读更多精彩内容