ubuntu16.04 安装pytorch (gpu/cpu)

首先聊一下:pytorch 比 tensorflow相对来说好安装些,而tensorflow 比 caffe 更加好安装。适用GTX1050,GTX1060,GTX1070,GTX1080等高级显卡系列安装好系统后,更新驱动。系统设置-》软件和更新-》附加驱动-》更新nvidia和intel安装cuda9下载 cuda9.run,(deb格式容易出错误)。执行如下命令:sudo sh cuda_9

首先聊一下:pytorch 比 tensorflow相对来说好安装些,而tensorflow 比 caffe 更加好安装。


适用GTX1050,GTX1060,GTX1070,GTX1080等高级显卡系列

  1. 安装好系统后,更新驱动。系统设置-》软件和更新-》附加驱动-》更新nvidia和intel
  2. 安装cuda9

下载 cuda9.run,(deb格式容易出错误)。执行如下命令:

sudo sh cuda_9.0.176_384.81_linux.run.81_linux-run


accept

n(不要安装driver)

y

y

y

安装完成后,设置环境变量。

设置环境变量:

ctrl+H 打开 (主目录/home/*/) .bashrc文件,添加

export PATH=/usr/local/cuda-9.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH 

##############注意 解决import tensorflow / pytorch 错误的问题############## 

更改为:(这里运行,不会再出现错误。很抱歉。)

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

终端运行:source ~/.bashrc

检查:nvcc --version


3. 安装cudnn7

下载 for linux版本的,执行如下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4. 安装anaconda(python3.5)

bash Anaconda3-4.2.0-Linux-x86_64.sh

yes(设置环境变量)

#################################

打开pytorch官网:

可以提前复制下载链接在win下,用迅雷下载pytorch;将安装包放到/home/*/下,pip install pytorch_name (也可以单独创建pytorch的环境:

conda create -n pytorch

source activate pytorch

pip install pytorch_name


再 pip install torchvision。 即可成功!

选择对应的版本,下面有安装命令,在终端 运行 命令 即可

##################################



作者:LEILEI18A
原文链接:https://blog.csdn.net/LEILEI18A/article/details/79871423

  • 发表于 2019-11-05 18:42
  • 阅读 ( 22 )
  • 分类:pytorch

0 条评论

请先 登录 后评论
不写代码的码农
LEILEI18A_csdn

0 篇文章

作家榜 »

  1. AI君 10 文章
  2. Tzung-Wen Liau 0 文章
  3. blairan 0 文章
  4. rookie 0 文章
  5. 陈凯 0 文章
  6. huanxue 0 文章
  7. admin 0 文章
  8. Lzs1998_csdn 0 文章