深度学习环境搭建1-pytorch框架

工作需要,自己实际动手安装的深度学习环境,包括CUDA、pytorch 、opencv等。

1 安装CUDA

1.1官网下载cuda

1.2安装深度学习环境搭建1-pytorch框架_第1张图片

 深度学习环境搭建1-pytorch框架_第2张图片

 深度学习环境搭建1-pytorch框架_第3张图片

1.3.配置环境变量

深度学习环境搭建1-pytorch框架_第4张图片

 深度学习环境搭建1-pytorch框架_第5张图片

 1.4.检查是否安装成功

深度学习环境搭建1-pytorch框架_第6张图片

2 安装cuDNN(7.6.5版本)

下载:cuDNN Archive | NVIDIA Developer

 深度学习环境搭建1-pytorch框架_第7张图片

 

上述三个都需要下载并按照顺序安装,否则会出错

安装命令:

sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb

sudo dpkg -i libcudnn7-dev_7.5.0.56-1+cuda10.0_amd64.deb

sudo dpkg -i libcudnn7-doc_7.5.0.56-1+cuda10.0_amd64.deb

深度学习环境搭建1-pytorch框架_第8张图片

 注意上图中第二个安装里面的cudnn.h的路径。

把此文件复制到/usr/local/cuda/include文件夹下面,并修改权限:

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

确认是否安装成功:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

深度学习环境搭建1-pytorch框架_第9张图片

 如上图所示,则表示已经安装成功

3 安装python(Anaconda)

深度学习环境搭建1-pytorch框架_第10张图片

 

4 搭建虚拟环境

conda create -n your_env_name python=3.6

激活虚拟环境:source activate your_env_name

关闭虚拟环境:source deactivate your_env_name

windows下激活:activate your_env_name

5 Linux下清华镜像安装pytorch

conda config --add channels Index of /anaconda/pkgs/free/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

conda config --add channels Index of /anaconda/pkgs/main/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

conda config --set show_channel_urls yes

conda config --add channels Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

先执行上边的语句,添加镜像,本次只添加了第一条速度就明显很快了

然后按照官网上下载对应版本的pytorch命令就可以了(失败了,更换了豆瓣源下载)

conda install pytorch torchvision torchaudio cudatoolkit=10.0 -c pytorch

下边用pip 使用豆瓣源安装成功了

pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html -i Simple Index

深度学习环境搭建1-pytorch框架_第11张图片

6 安装OpenCV-Python

1.下载离线包

2.进入到包所在路径,执行命令

conda install opencv3-3.1.0-py36_0.tar.bz2

 

你可能感兴趣的:(深度学习,pytorch,深度学习,人工智能,计算机视觉)