Windows 10系统在Anaconda下安装GPU版Pytorch

本文档介绍了在Windows 10系统中,使用Anaconda创建虚拟环境并安装GPU版PyTorch的详细步骤。首先确保CUDA和cuDNN已安装,接着创建名为'pytorch_gpu'的conda环境,安装Python 3.6。然后在环境中通过清华源安装匹配CUDA 10.0的PyTorch。最后,通过测试确保安装成功,并提供了处理CUDA版本不一致问题的方法。

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

在此之前默认Anaconda,CUDA, cuDNN 已安装好了。CUDA, cuDNN 下载对应版本安装。
我的环境为:win10+Anaconda+python3.6+cuda10.0

1、打开anaconda prompt
在这里插入图片描述
命令行输入:conda create --name pytorch_gpu python=3.6
python_gpu为anaconda下虚拟环境名称,可自定义,python=3.6为选择安装的python版本。
在这里插入图片描述
proceed选择y,回车, 等待相关包下载,可以看到在安装完成之后,信息提示:
在这里插入图片描述
如果要启用创建的环境,输入指令

conda activate pytorch_gpu

关闭该环境的话,输入指令

conda deactivate

2、安装pytorch

在刚刚创建的环境中安装pytorch, 在

评论 22
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值