Windows下安装PyTorch0.4.0

本文介绍了在Windows10系统中,使用Python3.5(Anaconda版)和NVIDIA GTX 960m GPU,如何通过pip安装PyTorch0.4.0和CUDA8.0、cudnn v6.0的详细过程。首先,创建并激活虚拟环境,然后通过pip下载指定版本的whl文件进行安装,最后测试CUDA、Tensors和MNIST,确保PyTorch正确运行。

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

Summary: Windows下安装Pytorch0.4.0
Author: Amusi
Date: 2018-04-25

注意: Pytorch0.4.0才开始支持Windows

Pytorch官网

PyTorch github



前言
PyTorch0.4.0开始支持Windows平台!官方支持!!!

环境介绍

硬件环境

  • NVIDIA GTX 960m


软件环境

  • Window10旗舰版
  • Python3.5(Anaconda版)
  • CUDA8.0
  • cudnn v6.0


安装教程

跟着官网走: http://pytorch.org/

安装方法1:pip

用pip安装,注意Python的版本和CUDA的版本(根据自己的配置选择)
1. 打开cmd命令提示符,先利用anaconda创建一个虚拟环境,命名为pytorch4
  • conda create -n pytorch4 python=3.5
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值