Mask CNN環境搭配與模型訓練中踩坑記錄

(一)搭建caffe2版 Caffe2已合併到pytorch框架裏了,所以搭建caffe2就得搭建pytorch。在搭建的過程中,各種坑,後來定位到 才知道,pytorch,要求cudnn7+,而cudnn7+要求對應的cuda在9+,然而,升級cuda,對整個服務器的影響較大,爲了不影響其他人的使用,放棄caffe2版本的搭建。所以在搭環境前,一定要看好需要的版本號。 (二)搭建tensorfl
相關文章
相關標籤/搜索