本文介绍了如何在环境中同时安装tensorflow-gpu 2.4.1和pytorch 1.7.1，包括cudatoolkit 10.1和cudnn 7.6.5。通过conda进行安装，详细阐述了tensorflow和pytorch的安装步骤，以及遇到的问题和解决方案。
想配置一个tensorflow+pytorch的环境，辛辛苦苦把tf配置好了结果pytorch又用不了了，真是按下葫芦起了瓢。
安装版本如下：
- tensorflow-gpu 2.4.1
- pytorch 1.7.1
- torchvision 0.8.2
- torchaudio 0.7.2
- cudatoolkit 10.1
- cudnn 7.6.5
Tensorflow
我用的是conda进行安装，貌似pip安装tensorflow的时候要自己下载各种依赖还有cudatoolkit，用conda相对来说一劳永逸一点。
conda install tensorflow-gpu
现在的tensorflow已经不区分cpu版和gpu版了，理论上来说conda install tensorflow也可以，但是貌似这样不会帮你安装cudatoolkit，所以还是老老实实加上-gpu的后缀吧。
这行命令会帮你安装好tensorflow、cudatoolkit、cudnn
验证一下：
import tensorflow as tf
import os
os.environ['CUDA_VISIBLE_DEVICES'
最低0.47元/天 解锁文章
1458
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
