1. 程式人生 > >[日常填坑]圖像分類實戰-服務器環境配置

[日常填坑]圖像分類實戰-服務器環境配置

linu nload vision 環境配置 ive blog his caffe 通過

服務器Ubuntu、pytorch框架、網絡模型SE-Resnet50,優化算法Adam

pytorch(python優先的深度學習框架,是一個和tensorflow,Caffe,MXnet一樣,非常底層的框架,它的前身是torch)

主要步驟(環境配置):

1. 需要下載cuda8.0

2. 需要下載pytorch(兩種方法)


註意:

第一種方式:可以下載Anaconda(一個python的包管理工具),通過conda下載pytorch

第二種方式:可以直接下載pytorch(如下命令行)

pip install 技術分享圖片http://download.pytorch.org/whl/cu90/torch-0.4.0-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install 技術分享圖片http://download.pytorch.org/whl/cu90/torch-0.4.0-cp27-cp27m-linux_x86_64.whl 註意:名字不要輸錯

第一種方式(下載的Anaconda也可以用於其他,很方便)

1. 安裝cuda8.0/cuda9.0

自行搜索教程(無坑)

2. 安裝Anaconda
wegt https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda2-5.1.0-Linux-x86_64.sh
cd download
ls
bash Anaconda2-5.1.0-Linux-x86_64.sh

# 然後一路enter,最後yes

#配置conda環境變量(一直按enter鍵,到最後有一個讓添加環境變量,你寫yes,如果沒有就需要手動添加)

http://www.cnblogs.com/gaofighting/p/8799169.html通過echo命令行手動添加環境變量
# 使用conda list查看安裝的包,測試是否正確安裝

# 置換鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

3. 安裝pytorch
conda install pytorch torchvision -c pytorch

註意:如果機子上安裝的cuda是9.0:conda install pytorch torchvision cuda90 -c pytorch

4. 驗證pytorch是否正確安裝
python
import torch

[日常填坑]圖像分類實戰-服務器環境配置