CycleGAN代码运行

CycleGAN代码运行,第1张

从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。

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

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

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

此时,我就成功安装了。

试用了一下不行

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/730059.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-27
下一篇 2022-04-27

发表评论

登录后才能评论

评论列表(0条)

保存