目录
为 PyTorch 创建虚拟环境
使用conda安装PyTorch (失败)
使用pip安装pytorch
安装cudatoolkit和cudnn
测试安装是否成功
首先,安装环境是:操作系统 Win10,已经预先安装了 Anaconda。关于 Anaconda 的安装步骤这里就忽略不讲了,Win10 下安装 Anaconda 非常简单。
为 PyTorch 创建虚拟环境安装 Anaconda 完毕后,我们在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中,我们通常会根据自己的需求去下载各种相应的框架库,但是可能每个项目使用的框架库并不一样,或使用框架的版本不一样,这样需要我们根据需求不断的更新或卸载相应的库,管理起来相当麻烦。所以通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。
总之就是想象成为 PyTorch 单独创建一个空间,这个空间与其它空间(如TensorFlow、其它版本的Python )互不影响。
创建 PyTorch 虚拟环境步骤:
1.打开 Anaconda 自带的 Anaconda Prompt:
2.打开 Anaconda Prompt 之后,在命令行输入:
conda create --name pytorch python=3.6
注意,这里的 pytorch 是虚拟环境的名称,可随意取。
3.然后继续下列命令,进入 pytorch 虚拟环境:
activate pytorch
效果如图所示:
好了,pytorch 的虚拟环境创建好了。接下来就是安装 PyTorch 了。
使用conda安装PyTorch (失败)
PyTorch 的官网提供了简单的安装方法,只需简单的命令即可。
首先,打开 PyTorch 官网安装页面:https://pytorch.org/get-started/locally/
我一开始是用的conda的方式安装,在下载pytorch1.4的安装包的时候1个小时都不动,一直是0%(如下图),遂放弃。
使用pip安装pytorch
然后我根据我自己的电脑环境,在pytorch官网中选择pip方式安装(如下图):
这样我们的pytorch就算安装好了。进入anaconda图形界面,可以看到在environment下,多了一个名叫“pytorch”的环境:
安装cudatoolkit和cudnn
首先看一下电脑CUDA的版本。控制面板——NVIDIA控制面板——帮助——系统信息——组件:
然后直接在anaconda的GUI中选择对应的版本依次安装即可:
测试安装是否成功
import torch
print(torch.cuda.is_available())
返回结果是True,则PyTorch和CUDA安装成功
参考资料:https://www.jb51.net/article/155635.htm
作者:游离丸子