1. 程式人生 > >小白pytorch+cycleGan踩坑筆記(GTX1060+ubuntu16.04)

小白pytorch+cycleGan踩坑筆記(GTX1060+ubuntu16.04)

因為之前沒有做過研究,從linux到cycleGan全部都是從零開始

踩坑第一次:在虛擬機器裡跑機器學習

結果總是顯示cuda版本和驅動版本不一致,後來終於發現,哦,虛擬機器裡沒顯示卡

裝雙系統 windows+ubuntu16.04

踩坑第二次,linux原驅動沒解除安裝乾淨

以為我還沒裝英偉達驅動=我沒裝驅動,所以直接裝英偉達驅動,結果總是顯示少個模組

最後將ubuntu完全重新安裝一遍重新開始

完整步驟

最後跑成功程式碼前進行的操作如下:

1、解除安裝原ubuntu驅動——2、安裝英偉達驅動——3、安裝cuda——4、安裝cudnn——5、安裝配置pip和python——6、安裝pytorch——7、安裝github中提到的依賴包——8、git程式碼——9、下資料集——10、執行程式碼

1-4 :

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

6-9:沒有遇到什麼問題

10:test的時候報錯了,去github上提問,作者說我是沒有更新最新的程式碼導致的