深度学习——Pytorch安装与配置(2)
Pytorch
- 没什么好说的,pytorch与tensorflow都是最常用框架
- 先安装anaconda,这个上篇文章说过了,
- 其次安装Cuda和Cudnn,这个之前做GPU时候说过了,cudnn的操作比较简单,找到正确的版本copy就好了
windows10安装
pytorch安装(基于Anaconda安装)
- 打开官网,官网可以自动识别你的系统、python以及cuda版本
- 打开Anaconda Prompt,输入命令:
conda create -n torch python=3.6
torch —— 替换成自己想要的名字
- 他会显示需要安装的包的名字和版本,当出现下面代码时,确定版本无误选择输入y
Proceed([y]/n)?
- 进入虚拟环境
conda activate torch
- 添加国内源,因为源是国外的,我们下载特别慢,可能会由于下载延时过高显示失败,这里需要添加国内源,指定每次下载显示源
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
- 安装pytorch
根据官网上面查找到的版本号及代码命令进行下载、
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2
- 安装done后再输入一次,确保所有的包安装完成,而且没有其他兼容问题,如有问题,请按照提示更新指定package
验证
- 当前虚拟环境下进入python环境
python
- 继续输入下列代码,无报错,输出下面结果即为安装成功(官网上可以看到)
import torch
x = torch.rand(5, 3)
print(x)
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
Anaconda检查
- 打开Anaconda Navigator
- 查看我们新建的虚拟环境
pycharm下使用该虚拟环境
-
打开pycharm(社区版免费,专业版破解方式,指路),新建project
-
添加环境,选择已存在环境
-
选择Conda Environment,然后选择我们刚刚建立的torch环境就好了
-
现在我们来验证一下
新建一个py文件,输入
import torch
x = torch.rand(5, 3)
print(x)
Run一下, 输出
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
查看库文件
本机运行
python -m pydoc -p 0
浏览器输入网址即可