Pytorch 安装使用介绍
Pytorch是一个由Facebook领导开发的端对端机器学习框架,其竞争者有名气很高的tensorflow模型。作为一个基于Python的框架,Pytorch致力于打造一个更快速和原生的机器学习体验[1]。
Pytorch 的安装
Pytorch可以通过conda或者pip直接安装。
(以下安装的全部为pytorch的最新稳定版本)
- 使用conda安装
-
conda install pytorch torchvision cudatoolkit=<CUDA Version> -c pytorch #With CUDA and GPU support conda install pytorch torchvision cpuonly -c pytorch #CPU