在同一环境中安装tensorflow和pytorch的GPU版
找到各自的一个版本,使得能共用同一版本的cuda,下面是一个可用的版本组合
NVIDIA GPU驱动: Driver Version: 440.44
conda install tensorflow-gpu=1.15.0
conda install pytorch torchvision cudatoolkit=10.0
找到各自的一个版本,使得能共用同一版本的cuda,下面是一个可用的版本组合
NVIDIA GPU驱动: Driver Version: 440.44
conda install tensorflow-gpu=1.15.0
conda install pytorch torchvision cudatoolkit=10.0