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

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

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

bert 的 tensorflow版预训练模型转pytorch版

pip install transformers

export BERT_BASE_DIR=pretrained-models/chinese_L-12_H-768_A-12
transformers-cli convert --model_type bert \
  --tf_checkpoint $BERT_BASE_DIR/bert_model.ckpt \
  --config $BERT_BASE_DIR/bert_config.json \
  --pytorch_dump_output $BERT_BASE_DIR/pytorch_model.bin

tensorflow与pytorch GPU版在同一环境中混合安
在同一环境中安装tensorflow和pytorch的GPU版找到各自的一个版本,使得能共用同一版本的cuda,...
CUDA版本与GPU驱动版本问题
GPU的驱动版本决定了支持CUDA的最高版本，所以安装tensorflow gpu（或者pytorch gpu）版...
tensorflow玩玩
安装：建议用conda安看自己的tensorflow是GPU还是CPU版： GPU版得服务器/电脑有显卡，需要安...
Tensorflow 模型训练
Tensorflow 环境搭建 Windows GPU 版安装依赖软件包 Tensorflow 1.5.0/1....
PyTorch安装
机器已经安装过tensorflow了，GPU相关环境都装过了访问 https://pytorch.org/get-...
2019-04-09
anaconda安装pytorch，tensorflow-gpu环境。前提：笔记本英伟达显卡驱动已经安装好了。 ...
一文上手最新Tensorflow2.0系列(五)｜使用GPU加速
1. 安装配置GPU环境 1.1. 安装GPU版TF 在之前我们已经安装了CPU版的TensorFlow，为了使用...
conda 下的TensorFlow与keras
tensorflow 简单利用conda安装tensorflow-gpu=2.2.0虚拟环境部分不再赘述，直接给安...
查看版本号命令
conda版本 GPU版本 CUDA版本 pytorch版本验证GPU版pytorch是否可用 torchvis...
服务器GPU使用那些事
持续更新中 GPU与tensorflow的那些事要装TensorFlow with GPU support，有以...