在同一个conda环境下安装tensorflow和pytorch

Posted 2023-03-18

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了在同一个conda环境下安装tensorflow和pytorch相关的知识，希望对你有一定的参考价值。

参考技术A

为啥要装在一个环境里呢，比如我256G固态的笔记本硬盘空间珍贵，比如实验室里公用的服务器一个人只允许新建一个conda虚拟环境，这里就是找到一个兼容的版本，也没啥好啰嗦的。
首先打开 nvidia-smi 确保右上角cuda版本大于 10.0 。

如果想找其它兼容的版本怎么弄呢？

先用nvidia-smi确定本机cuda版本，比如是 11.0

确保这两个 cudatoolkit 的版本相同并且 小于nvidia-smi显示的版本 应该就可以了。

再比如

以上是关于在同一个conda环境下安装tensorflow和pytorch的主要内容，如果未能解决你的问题，请参考以下文章