CycleGAN代码运行

从github上clone了一个cycleGAN项目,试着在服务器上运行一下!

https://github.com/Lornatang/CycleGAN-PyTorch

由于代码比较久远,需要安装版本比较低的pytorch才能运行。

我安装的版本pytorch0.4.1,cuda 11.1

我们可以先将需要的包下载下来,然后直接离线安装,Pytorch各个版本的下载地址可以点[这里]
http://download.pytorch.org/whl/torch_stable.html

这个是安装的cuda9.2+pytorch0.4.1+python3.6+linux版本的pytorch。

CycleGAN代码运行_第1张图片

然后需要安装torchvision,安装torchvision0.10+cuda11.1(我需要的是cuda11.1,这样子安装会导致torch0.4.1被uninstalled)

CycleGAN代码运行_第2张图片

所以我需要再安装一下torch0.4.1,此时会显示torch0.4.1与torchvision不兼容。

此时再 conda install torchvision -c pytorch(这个是从官网下,如果下载很慢,可以把-c pytorch删掉,这样就是用镜像下载就会快一点)。

此时,我就成功安装了。

试用了一下不行

你可能感兴趣的:(机器学习)