PyTorch在64位Windows下的Conda安装

最新推荐文章于 2025-06-18 14:11:04 发布

心跳的吵闹

最新推荐文章于 2025-06-18 14:11:04 发布

阅读量1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签： python 框架 64位 windows

本文链接：https://blog.youkuaiyun.com/u014533320/article/details/78639638

深度学习专栏收录该内容

3 篇文章

订阅专栏

本文介绍如何在Windows 10环境下使用Anaconda3安装Pytorch，并确保CUDA和cuDNN正常运行。文章提供了具体的安装命令及测试代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天，看一篇paper中用的是pytorch实现，所以安装一下这个框架，以便以后学习，由于之前安装过tensorflow，所以此次安装pytorch很顺利，省去了不少麻烦。

在win10 python3.6开启Pytorch支持：

你需要满足以下条件：

Anaconda3 (with Python 3.5/3.6)
Windows 64位系统（Windows 7 或 Windows Server 2008 及以上）
CUDA 8.0.61
cuDNN v5以上

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

conda install -c peterjc123 pytorch_legacy

（我是科学上网的，下载速度还可以）

由于我在之前安装了tensorflow，所以省去了安装cuda,cudnn.

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

哈哈，大功告成！