Pytorch是在Torch基础上开发的,其与Torch 的不同之处在于PyTorch 使用了Python 作为开发语言 所谓“Python first”,说明Pytorch是以Python 优先的深度学习框架,其不仅能够实现强大的GPU 加速,同时还支持动态神经网络,这是现在很多主流框架比如Tensorflow 等都不支持的。
1、Window10下安装Anaconda3(64bit)
Anaconda指的是一个开源的Python发行版本,其包含了conda、Python等180多个科学包及其依赖项。里面所包含的Jupyter Notebook是数据挖掘领域中最热门的工具。(例如Kaggle网站)
若安装时未勾选添加环境变量,则需手动添加,方法如下:
假设Anaconda3的安装路径为:D:Anaconda3 则Anaconda3Scripts的路径为:D:Anaconda3Scripts Anaconda3Libraryin的路径为:D:Anaconda3Libraryin
Spyder是随anaconda一起安装的IDE,在Spyder中输入print('hello')验证是否安装成功
2、在anaconda里安装pytorch
(1)创建pytorch虚拟环境
在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中,我们通常会根据自己的需求去下载各种相应的框架库,但是可能每个项目使用的框架库并不一样,或使用框架的版本不一样,这样需要我们根据需求不断的更新或卸载相应的库,管理起来相当麻烦。所以通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。总之就是想象为 PyTorch 单独创建一个空间,这个空间与其它空间互不影响。例如安装TensorFlow的时候也是创建了一个虚拟环境。
方法一:Anaconda Navigator-Environments。如果之前没有创建过任何虚拟环境,这里只会有一个base(root)。点击下方的create,创建一个叫做pytorch的虚拟环境。
方法二:打开anaconda prompt,输入相应命令conda create --name pytorch python=3.7(版本号自行匹配),安装成功后可以在Anaconda Navigator-Environments的base(root)下看到pytorch
(2)激活虚拟环境
打开anaconda prompt,可以看到行首是(base),表明此时处于base环境中,输入activate Pytorch,可以看到(base)变为了(pytorch),说明进入Pytorch环境。
(3)安装pytorch
输入在
注:查看cuda版本号方法:打开CMD,输入nvcc --version
即可
(4)验证pytorch的安装