码迷,mamicode.com
首页 > 其他好文 > 详细

PyTorch----安装

时间:2018-02-07 14:59:08      阅读:1368      评论:0      收藏:0      [点我收藏+]

标签:linux   ack   type   art   ice   gpu   efault   and   位置   

 由于PyTorch官网 http://pytorch.org/ 推荐使用Anaconda作为软件包管理器(package manager),所以我们需要首先准备好Anaconda环境。

1.准备好Anaconda环境

具体参见:http://blog.csdn.net/zhdgk19871218/article/details/46502637

2.进入PyTorch官网,选择自己相应的版本

 查看python版本

1 fenglei@gpu01:~$ python -V
2 Python 3.5.2 :: Anaconda 4.2.0 (64-bit)

查看cuda版本

1 fenglei@gpu01:~$ cat /usr/local/cuda/version.txt
2 CUDA Version 8.0.61

注意:确保拥有最新的pip 和 numpy packages.

1 fenglei@gpu01:~$ pip install --upgrade pip
2 Collecting pip
3   Downloading pip-9.0.1-py2.py3-none-any.whl (1.3MB)
4     100% |████████████████████████████████| 1.3MB 13kB/s
5 Installing collected packages: pip
6   Found existing installation: pip 8.1.2
7     Uninstalling pip-8.1.2:
8       Successfully uninstalled pip-8.1.2

 

1 fenglei@gpu01:~$ pip install --upgrade numpy
2 Requirement already up-to-date: numpy in ./anaconda3/lib/python3.5/site-packages

3.根据Run this command的提示,安装PyTorch

技术分享图片

如果下载速度太慢的话,在自己的Linux系统上可以在浏览器里新建任务,下载链接就是安装命令中第一行的网址,把这个whl文件下载下来;在Windows上可以用迅雷下载,新建任务进行下载。

下载完成之后拷贝到自己的Linux系统上去或者服务器上,然后找到文件位置,直接pip install 文件名

4.验证PyTorch是否安装成功

1 fenglei@gpu01:~$ python
2 Python 3.5.2 |Anaconda 4.2.0 (64-bit)| (default, Jul  2 2016, 17:53:06)
3 [GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux
4 Type "help", "copyright", "credits" or "license" for more information.
5 >>> import torch
6 >>> import torchvision
7 >>>

不报错的话就说明安装成功。

 

PyTorch----安装

标签:linux   ack   type   art   ice   gpu   efault   and   位置   

原文地址:https://www.cnblogs.com/smile-sunshine/p/8425735.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!