本文档介绍了如何在anaconda环境下同时安装pytorch和tensorflow-gpu，强调了两者对cuda版本的依赖。首先，安装cuda版本为10.1的tensorflow-gpu，然后通过cuda版本匹配相应的pytorch版本进行安装。安装完成后，通过python代码验证gpu是否可用，并检查pytorch的安装。该过程旨在提供一种解决多库共存的方法。
anaconda下同一个虚拟环境安装pytorch和tensorflow
最近在重装一个硬盘，为了方便，希望能在一个conda环境下使用pytorch和tensorflow-gpu，由于pytorch和tensorflow-gpu都对cuda的版本有不同的对应关系，需要寻找相同cuda版本下的pytorch和tensorflow-gpu。例如：
conda install tensorflow-gpu=2.4.1
conda命令能装的目前貌试最高到2.4.1，对应的cuda为10.1.安装完成后进入环境,检查是否可用。
import tensorflow as tf
tf.test.is_gpu_available()
然后，根据其cuda版本寻找对应的pytorch
如图，选择第二条的版本
# conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
输入对应指令完成安装。检测pytorch是否安装成功
2万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
