从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删掉,这样就是用镜像下载就会快一点)。
此时,我就成功安装了。
试用了一下不行
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)