我终于跑通StyleFlow了

最新推荐文章于 2024-04-25 09:40:49 发布

RICHPAPA

最新推荐文章于 2024-04-25 09:40:49 发布

阅读量926

点赞数 2

分类专栏： cv 文章标签：人工智能 pytorch tensorflow ubuntu 计算机视觉

本文链接：https://blog.csdn.net/qq_41694777/article/details/121038810

版权

cv 专栏收录该内容

2 篇文章 1 订阅

订阅专栏

本文记录了配置StyleFlow项目的全过程，包括在RTX3060和RTX6000上搭建cuda10.0和cuda11.1环境，详细介绍了安装Tensorflow1.15.0和PyTorch1.8.0的步骤，以及解决过程中遇到的坑。特别指出，cuda11.1环境下安装的torch可用于cuda10.0环境。此外，还探讨了如何在不同GPU上运行StyleFlow的问题。

摘要由CSDN通过智能技术生成

我终于把StyleFlow跑通了！又pytorch又tensorflow的烦死了，踩坑踩得我要吐了！在这里记录一下整个过程防止我以后忘记，欢迎沟通交流呀！

GitHub项目地址：https://github.com/RameenAbdal/StyleFlow

他给了一个requirements.txt,按理来说按照这个配也行，但是可能还是有各种各样的问题，话不多说，我们开始吧！

我的两个亲测成功的环境：

1.是我自己的电脑，显卡是RTX3060，众所周知cuda10不支持3060，所以是cuda11.1+tensorflow-gpu1.15.0+torch1.8.0。

2.是实验室的服务器，显卡是RTX6000，环境是cuda10.0+tensorflow-gpu1.15.5+torch1.8.0。

都是python3.7，Ubuntu18.04。

下面开始配环境：

首先安装对应cuda、cudnn，/usr/local目录下建立好cuda的软链接，然后按照github上的步骤：

git clone https://github.com/RameenAbdal/StyleFlow.git
cd StyleFlow/
conda env create -f environment.yml

会创建StyleFlow的环境，但是可能会由于种种原因报点小错。。

conda activate StyleFlow

进入StyleFlow环境下：

python main.py

可能会提醒你这个包没安，那个包没安，如matplotlib、opencv呀等等，不要慌张，缺什么就pip install什么或者conda install什么，如果出现没有pytorch或tensorflow没有的情况就按照我上面说的版本安装就好了。

值得一提的是，cuda10.0和cuda11.1的环境我安装的都是pip安装的torch1.8.0+cu111,服务器的环境本来是cuda11.1，我先安装的pytorch，然后又把cuda11.1换成的cuda10.0，用torch.cuda.is_available()一看竟然也是true，我就没重新安装pytorch。

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

StyleFlow和stylegan2环境要求相同，必须是tensorflow1.x，但tensorflow1.x只支持cuda10.0及以下的版本，github上的issues有人提到过tensorflow2.x的解决方法，但说的不详细，我也没尝试成功。

服务器上的tensorflow我是用conda安装的，cuda10.0直接安装就能成功，自己3060的电脑上我是用pip安装的，由于是cuda11.1，所以用conda安装可能会报环境冲突的错，于是我从pip install tf1.1开始试，每安一次就tf.test.is_gpu_available()，直到tf1.15.0报了true，但是我在服务器另一块显卡RTX3090上试tf1.x，就没有成功过，所以目前我还不能用3090。