PyTorch在64位Windows下的Conda包

最新推荐文章于 2024-08-11 21:26:28 发布

钟子悦

最新推荐文章于 2024-08-11 21:26:28 发布

阅读量4.5k

点赞数

分类专栏： Pytorch 文章标签： pytorch

Pytorch 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

原贴来自知乎：https://zhuanlan.zhihu.com/p/26871672

安装pytorch也是跌跌撞撞的，现在终于弄好了。

更新：已经添加了对所有Compute Capability>=2.0的显卡和老版本Windows的支持

更新：已经支持最新的0.2版本，重新键入安装命令即可进行安装或升级

先别急着激动。如果要直接使用的话，你需要满足以下条件：

Anaconda3 (with Python 3.5/3.6)
Windows 64位系统（Windows 7 或 Windows Server 2008 及以上）
CUDA 8.0.61（包中已内置，因此老版本也能用）
cuDNN v5以上（包中已经自带了v6, cuDNN v5用户亦可放心使用）

这四个条件个人感觉还算比较OK，如果不想放弃Anaconda2也可以创建虚拟环境来使用。

要安装的话，如果你不嫌弃anaconda cloud的网速的话，只需根据自己的系统键入下面的一条命令即可：

# for Windows 10 and Windows Server 2016
conda install -c peterjc123 pytorch

# for Windows 7/8/8.1 and Windows Server 2008/2012
conda install -c peterjc123 pytorch_legacy

如果不能忍受conda那蜗牛爬般的网速的话，那我为大家将包上传至公有云中，可以尝试百度云，大家下载之后，键入如下几条指令：

conda install numpy mkl cffi
conda install --offline path/to/tar/pytorch????.tar.bz2

注：文件名说明：

一般为PACKAGENAME-VERSION-PYTHON_VERSIONcu80.tar.bz

PACKAGENAME 分为 pytorch 和 pytorch_legacy，分别为NT内核版本10和6的两类系统进行编译；VERSION 代表 pytorch 的版本；而PYTHON则代表python程序的版本，主要分为3.5和3.6；请自行选择合适的版本进行安装。

安装之后，也千万要注意，要在主代码的最外层包上

if __name__ == '__main__':

这个判断，可以参照我昨天文章中的例子，因为PyTorch的多线程库在Windows下工作还不正常。

更新：经网友提醒，若import torch时发生如下错误：

Traceback (most recent call last):
  File "test.py", line 2, in <module>
    import torch
  File "C:\Anaconda3\lib\site-packages\torch\__init__.py", line 41, in <module>
    from torch._C import *
ImportError: DLL load failed: The specified module could not be found.

请将Anaconda的Python版本从3.6.0升级至3.6.1。

附一段简单测试CUDA与cuDNN是否工作正常的代码：

# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))