小白pytorch+cycleGan踩坑笔记(GTX1060+ubuntu16.04)

因为要做项目,所以需要复现 https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix 

因为之前没有做过研究,从linux到cycleGan全部都是从零开始

踩坑第一次:在虚拟机里跑机器学习

 

结果总是显示cuda版本和驱动版本不一致,后来终于发现,哦,虚拟机里没显卡

小白pytorch+cycleGan踩坑笔记(GTX1060+ubuntu16.04)_第1张图片

装双系统 windows+ubuntu16.04

 

关于双系统,这篇文章讲的很清楚:https://blog.csdn.net/fesdgasdgasdg/article/details/54183577

踩坑第二次,linux原驱动没卸载干净

 

以为我还没装英伟达驱动=我没装驱动,所以直接装英伟达驱动,结果总是显示少个模块

最后将ubuntu完全重新安装一遍重新开始

完整步骤

最后跑成功代码前进行的操作如下:

1、卸载原ubuntu驱动——2、安装英伟达驱动——3、安装cuda——4、安装cudnn——5、安装配置pip和python——6、安装pytorch——7、安装github中提到的依赖包——8、git代码——9、下数据集——10、运行代码

1-4 :

看的文档是https://blog.csdn.net/u010801439/article/details/80483036 写的很详细很好

5:ubuntu16.04里面本来就有python3.5,所以没有另外安装,但是为了使用方便,将python3的软连接修改为python(也就是当在命令行输入python的时候调用的是python3.5不是python2.7),修改的wiki看的是 https://blog.csdn.net/m0_37639542/article/details/78311795

另外pip默认的是python2的,所以我也参考python修改软链接的方式将pip的软链接也修改了

还有就是,无论是pip2还是pip3都需要将pip更新到最新版本,但是pip3更新的话可能会遇到问题

Traceback (most recent call last): 
File “/usr/bin/pip3”, line 9, in 
from pip import main

解决方式参考的文档: https://blog.csdn.net/cow66/article/details/80069309

6-9:没有遇到什么问题

10:test的时候报错了,去github上提问,作者说我是没有更新最新的代码导致的

你可能感兴趣的:(小白pytorch+cycleGan踩坑笔记(GTX1060+ubuntu16.04))