ubuntu PyTorch 安装

来源：互联网发布：淘宝索尼z2手机壳编辑：程序博客网时间：2024/06/05 04:17

----------------运行anaconda-----
bash Anaconda...sh

----用python3.6替换python2.7-------------
sudo ln -s /usr/local/bin/python3.6 /usr/bin/python

-----------添加镜像下载加快----------------
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

----------安装显卡驱动---------------------
装机时已经安装了，不再安装驱动

---------------安装cuda-------------------
sudo sh cuda_8.0.61_375.26_linux.run
==========点击顺序==============
accept
n 显卡驱动不装
n openGL不装
y
y
....
=================================
cuda被安装在 /usr/local 目录下

---------添加环境变量--------------------
sudo vim ~/.bashrc
打开文件将下面两行添加到末尾
export LD_LIBRARY_PATH="$LD_LABRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

-----------------安装cudnn-----------------
dpkg -i libcudnn7_7.0.5.15-1+cuda8.0_amd64.deb

---------------安装pytorch---------------

conda install pytorch torchvision -c soumith

------------测试-----------------------

$python

>>>import torch

>>>torch.cuda.is_available()

True

阅读全文

0 0