windows下pytorch的安装

最新推荐文章于 2025-10-22 16:30:53 发布

转载最新推荐文章于 2025-10-22 16:30:53 发布 · 755 阅读

文章标签：

#pytorch的安装

pytorch 专栏收录该内容

8 篇文章

订阅专栏

本文提供了详细的PyTorch安装步骤，包括下载适用于Windows系统的conda包、安装过程中的具体命令及如何验证CUDA与cuDNN是否正确配置。同时，还提供了解决import torch时可能出现的问题的方法。

部署运行你感兴趣的模型镜像

PyTorch的安装

1.下载windows版本的conda包

conda包下载地址：https://pan.baidu.com/s/1dF6ayLr#list/path=%2Fpytorch。大家根据需要选择相应的文件下载。

注：文件名的解释：一般为PACKAGENAME-VERSION-PYTHON_VERSION cu80.tar.bz

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

2.安装conda包

打开命令行提示符，cd到conda包下载的位置。然后运行下面两个命令：

conda install numpy mkl cffi
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar

3.测试PyTorch

附一段简单测试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))

如果能正常输出结果则表明安装成功。

4.import torch出错处理

如果时原来已经安装了anaconda3，在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.

这时我们只需要更新一下Anaconda3的python版本至3.6.1及以上即可。更新的方式有以下两种：

conda install -c anaconda python=3.6.1 # 更新至指定版本
conda updata python # 直接更新

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理