同时安装 Tensorflow&Pytorch

Pytorch 与 Tensorflow 可一同安装,但需要注意版本对应问题,找到一个两者都能用的CUDA号,
如:CUAD 11.0; Pytorch v1.7.1; tensorflow_gpu-2.4.0; Keras 2.4.3; Python 3.6-3.8; 推荐 cuDNN8.0

流程:
1.确定tensorflow与pytorch的版本,进而确定python、CUDA等环境的版本;
2.安装VC++相关软件包(最新支持Visual C++ 可再发行程序包下载 | Microsoft Docs);
3.卸载python中的numpy(pip uninstall numpy)(每个tensorflow都有相对应的numpy版本,避免冲突先卸载);
4.安装对应版本的CUDA与cuDNN并检验;
5.安装对应版本的tensorflow(pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple),并在python中运行检验(import tensorflow as tf tf.test.is_gpu_available() );
6.安装对应版本的Keras;
7.安装对应版本的Pytorch(pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html),并检验安装成功(import torch torch.cuda.is_available() );

Original: https://blog.csdn.net/holly_poppy_/article/details/123092930
Author: holly_poppy_
Title: 同时安装 Tensorflow&Pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496959/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球