天天看點

linux安裝pytorch

1、官網下載下傳安裝anaconda3

2、root環境預設的python版本

linux安裝pytorch

檢視root環境資訊

linux安裝pytorch

更新了一下conda版本

linux安裝pytorch

3、建立一個新環境,以此來根據需求安裝包,友善環境管理與維護

linux安裝pytorch

激活并進入新環境

linux安裝pytorch

檢視目前環境已安裝的包

linux安裝pytorch

4、安裝jupyter notebook

linux安裝pytorch

5、添加清華鏡像源,提高pytorch的安裝速度

linux安裝pytorch

6、注意:安裝pytorch的前提是,如果伺服器有顯示卡,需要安裝cuda。見pytorch官網。

檢視伺服器的cuda版本

linux安裝pytorch

注意:/usr/local/cuda是一個軟連結l,連結到要使用的cuda版本,類似于windows下的快捷方式

linux安裝pytorch

安裝cuda版本對應的pytorch,torchvision

linux安裝pytorch
linux安裝pytorch

7、退出目前環境并進入root環境

linux安裝pytorch

重新進入

linux安裝pytorch

此時驗證pytorch是否安裝成功

linux安裝pytorch
linux安裝pytorch

8、附加:

主機名太長,修改一下

linux安裝pytorch
linux安裝pytorch
linux安裝pytorch

删除時最好加個參數-i,多一個提示,詢問是否删除

linux安裝pytorch

繼續閱讀