win10安装pytorch-gpu

本文详细介绍了如何安装和配置CUDA11.3,包括环境变量设置,并指导如何在conda环境下创建新环境,安装PyTorch。特别地,针对conda列表中显示PyTorch为CPU版本的问题,提供了手动下载特定版本的whl文件并使用pip安装的解决方案。最后,还演示了如何安装jupyter notebook并在新环境中启动。
摘要由CSDN通过智能技术生成

步骤总结:

1. 下载cuda toolkit11.3, 完成环境配置path,安装后cmd窗口输入nvcc -V查看安装信息

2. 下载cudaa 11.x, 解压缩到cuda11.3覆盖三个文件夹

3. conda create -n mytorch python=3.7 创建新环境,装过cpu版的一定要在新环境中重装.

4. 按照其他教程安装验证为True的就结束。这里写False的解决方法,针对conda list查看pytorch是cpu版本的情况(可能和源有关)——

https://download.pytorch.org/whl/torch_stable.html手工下载电脑对应版本的三个whl包,pytorch官网自动显示的命令也是这三个包:

使用pip install 路径\文件名  分别进行安装

5. pip install jupyter ,下载jupyter notebook到新环境,命令窗口输入jupyter notebook启动

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值