深度学习学习笔记(一):pytorch及tensorflow2.0安装

先检查环境,看一看电脑cuda的版本(windows参考https://www.jianshu.com/p/d3b9419a0f89,ubuntu参考https://blog.csdn.net/qq_16525279/article/details/80662217)
安装cuda及cudnn:https://zhuanlan.zhihu.com/p/65557545

conda

一、常用命令:

conda list     查看安装的包
conda env    查看存在哪些虚拟环境
conda update conda      更新conda
conda update --all       更新所有的包
conda remove package_name     删除某个包

二、更换清华源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes

三、虚拟环境:

创建虚拟环境:conda create -n your_env_name python=X.X,其中your_env_name自己起一个好记的名字即可。可按anaconda->envs路径找到此环境。
启动虚拟环境:conda activate your_env_name
关闭虚拟环境: 在该虚拟环境下conda deactivate
删除虚拟环境: conda remove -n your_env_name --all

pytorch

一、登陆pytorch官网:

https://pytorch.org/get-started/locally/
深度学习学习笔记(一):pytorch及tensorflow2.0安装_第1张图片

二、使用pip安装:

直接复制上图中的pip3 install …,然后将pip3改为pip
深度学习学习笔记(一):pytorch及tensorflow2.0安装_第2张图片
这样安装有时候会很慢。。。下面添加清华源:


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

然后再安装:conda install pytorch=1.2.0 torchvision=0.4.0

三、测试是否可用:

import torch
print(torch.__version__)
print(torch.cuda.is_available())
> '1.2.0'
> True

tensorflow2

一、安装

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn tensorflow-gpu==2.0.0

二、检测是否可用

import tensorflow as tf
print(tf.__version__)
> '2.0.0-beta1'
print(tf.test.gpu_device_name())

输出如图:
深度学习学习笔记(一):pytorch及tensorflow2.0安装_第3张图片

你可能感兴趣的:(深度学习)