PyTorch是一个基于Torch的Python开源机器学习库,用于 自然语言处理等应用程序。它主要由Facebookd的人工智能小 组开发,不仅能够实现强大的GPU加速,同时还支持动态神经 网络,这一点是现在很多主流框架如TensorFlow都不支持的。 PyTorch提供了两个高级功能:
◈ 具有强大的GPU加速的张量计算(如Numpy)
◈ 包含自动求导系统的深度神经网络
在Linux系统中需要先下载好conda,
- 创建python3.7环境: $ conda create -n kg python=3.7
- 激活环境: $ conda activate kg
- 安装需要的tqdm和sklearn包 conda install tqdm pip install sklearn
在网址中选择合适的Pytorch安装
https://pytorch.org/get-started/locally/
下载过慢的情况需要先输入下面指令添加源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后复制官网命令,将最后的-c pytorch去掉,如: