pytorch 在linux採用離線方式下載,軟件包 安裝 torch,torchvision,並對應版本,以及cpu,gpu包的問題

1、 pytorch linux 下的安裝教程

https://blog.csdn.net/xiangyuel/article/details/94969820?utm_medium=distribute.pc_relevant.none-task-blog-baidujs-1

2、 下載地址,版本對應

torch, torchivision, 版本對應
torch, https://pypi.org/project/torch/1.5.0/#files,找到需要的版本下載
torchvision , https://pypi.org/project/torchvision/#files ,安裝包下載地址
下載後用 pip install 安裝
下載包一定要知道對應關係,版本 torch torchvision , 查看兩個包的發佈時間,
pip 安裝自動匹配版本關係,知道版本對應

pip install torch-1.5.0-cp36-cp36m-manylinux1_x86_64.whl ,
pip install torchvision-0.6.0-cp36-cp36m-manylinux1_x86_64.whl 這是對應的

在這裏插入圖片描述

在這裏插入圖片描述torchvision 可以早發佈早幾天

3、上述吭哧吭哧搞定,發現不能GPU 訓練,原來上述都是cpu 版本的離線安裝,重要事情說三遍:cpu軟件包,不是gpu軟件包,gpu軟件包下載長這樣

在這裏插入圖片描述
torch.cuda.device_count() 一直0
torch.cuda.is_available()返回false

https://download.pytorch.org/whl/torch_stable.html
官網地址,找到 cu開頭的包,就是cuda,
下載的界面如下,版本的選擇按照第2個步驟發佈日期 選擇對應版本即可,
下載結束直接pip install 安裝包, 如果沒有版本問題,馬上就會顯示 Successfully 信息

在這裏插入圖片描述

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