在win10上安装pytorch-gpu版本

本文档介绍了如何在Windows 10上利用Anaconda安装PyTorch-GPU版本。首先,参照TensorFlow-GPU的安装教程安装Anaconda。接着,通过conda命令创建名为pytorch-gpu的虚拟环境,并从PyTorch官网获取对应安装包。如果下载速度慢,可以添加镜像源。安装完成后,验证Python环境中PyTorch的成功导入。最后,在PyCharm中配置该Anaconda虚拟环境作为Python解释器,实现PyTorch-GPU在PyCharm上的使用。
摘要由CSDN通过智能技术生成

目录

 1.1  anaconda安装

 1.2  pytorch-gpu安装


1.1  Anaconda安装

anaconda的安装请看我之前发的tensoflow-gpu安装,里面有详细的安装过程,这里不做重复描述,传送门

1.2  pytorch-gpu安装

anconda下载完成后,打开anaconda prompt,输入

conda create -n pytorch-gpu python=3.7 创建pytorch用的虚拟环境,当然你们虚拟环境也可以不用命名成pytorch-gpu,创建好后,在pytorch官网找到对应的安装包就行下载,我这里直接

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

如果安装太慢就需要加上镜像源了

永久生效 (pip config set global.index-url+国内镜像地址)
pip config set gl
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

bug生成中

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值