百度AI Studio配置並運行pytorch項目的體驗

總之,體驗我給7分(滿分10分)。

優點:

  • GPU挺快的,當然這一點沒有比較過,直觀感受上還挺快的

  • 算力卡(GPU可用時長)給得跟不要錢似的,完全就是那種屠龍寶刀來就送的感覺,而且我這算力卡怎麼越用越多……

缺點:

  • notebook裏只能使用百度家的飛槳框架(雖然在命令行可以用其他框架,但不穩定)
  • 動不動掉線,一掉線裝好的依賴就全沒了……在網上看到一堆持久化的方法,但好像並沒有用,我沒有往持久化的目錄裏寫數據的權限,也沒法sudo。要是有什麼好辦法,請各位指教~~,然後我把評分改高一點~~。

因爲它的缺陷特別敗好感,所以我只能給個7分。


剛纔談到,他這個平臺不能在notebook裏運行別的框架,所以pytorch只能在命令行用。還好他裝了3.7的py,可以用pip(沒有conda)安裝pytorch。但是在裝之前,先換成清華的鏡像,這樣能快一點:

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

然後看一下平臺GPU的版本,免得裝錯了pytorch的版本:

nvcc -V
# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2018 NVIDIA Corporation
# Built on Tue_Jun_12_23:07:04_CDT_2018
# Cuda compilation tools, release 9.2, V9.2.148

然後查看官方文檔,因爲這裏是9.2的cuda,所以需要這麼安裝:

pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

如果裝錯了,運行的時候就會報錯。會出現這個情況:

torch.cuda.is_available() = False

感覺上相當於就是驅動不匹配,沒法使用GPU,所以一定要裝對。

等一會,安裝完了之後,在命令行輸個python,然後import torch試試,如果可以正常運行,那就正式完成了,可以開始使用。不過,只能在命令行用,即使是裝好了,也不能在notebook裏用:

Cannot run import torch because of system compatibility. AI Studio prepared an entire environment based on PaddlePaddle already. Please use PaddlePaddle to build your own model or application.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章