Win10+Cuda10.2的Pytorch安装

Win10+Cuda10.2的Pytorch安装


今天是2022年10月底,最新版本Cuda稳定版已不支持Cuda10.2
由于电脑是非安培架构(30系)的NV的GPU,故装了Cuda10.2

(如果你使用的是安培架构的GPU,推荐使用CUDA11以上。如果你使用的是非安培架构的GPU,推荐使用CUDA10.2,性能更优。)
——百度Paddle

安装Cuda和Cudnn加速库的教程随便搜一下CSDN教程即可。以下教程从装好Anaconda和cuda10.2 以及cuDNN 8.2.1开始:

1. 建立一个深度学习的环境,以免一个崩了以后影响所有的包
进入anaconda promote 命令行,先把conda 包都升级一下

conda update conda 
conda update --all -y

随后建立一个机器学习的环境,Python版本指定为3.8好了。

conda create --name pytorch python==3.8

2. 开始装包
打开windows搜索栏,可以看到一个叫Anaconda Navigator (Anaconda3)的绿色环为图标的软件,可以比较方便的管理环境和包。

官网的最新版本pytorch稳定版已不再支持cuda 10.2的了,所以我们要去下老版本
在这里插入图片描述
但是老版本里很多是只支持cpu版本不支持GPU版本的,所以需要您去
https://download.pytorch.org/whl/torch_stable.html
里寻找。
直接说结论
cuda10.2最新GPU版本支持到
torch1.10.0+cu102 torchvision0.11.0+cu102 torchaudio==0.10.0
下载地址:
https://download.pytorch.org/whl/torch_stable.html
文件名分别是:
torch-1.10.0+cu102-cp38-cp38-win_amd64
torchvision-0.11.0+cu102-cp38-cp38-win_amd64
torchaudio-0.10.0+cu102-cp38-cp38-win_amd64

这三个文件下载到本地后,就需要安装了。
请注意若您在Anaconda Promote里直接输入 pip install xxxx.whl的话,默认是安装到(base)而非您的机器学习环境(pytorch)里
故需要使用以下命令安装

activate pytorch #进入我创立的机器学习环境
where python #找到python 文件在哪
C:\Users\admin\Anaconda3\envs\pytorch\python -m pip install xxxxx

注意:1. 上述C:\Users\admin\Anaconda3\envs\pytorch\python 为您使用where python的输出,告诉命令行我要把这个包装到我这个pytorch的python环境而非base环境
2. 后面的xxxx 需要您把下载下来的文件拖到命令行里

比如我的命令就是这样:

C:\Users\Castor_WONG\Anaconda3\envs\paddle\python -m pip install C:\Users\Castor_WONG\Downloads\torch-1.10.0+cu102-cp38-cp38-win_amd64.whl

因为我的机器学习环境python文件路径在:
C:\Users\Castor_WONG\Anaconda3\envs\paddle\python
我下载的torch文件在
C:\Users\Castor_WONG\Downloads\torch-1.10.0+cu102-cp38-cp38-win_amd64.whl

回车安装,等待。进入anaconda promote 依次输入python ,import torch,print(torch.cuda.is_available()) 变成True即安装成功

  • 4
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
对于安装 CUDA 10.2PyTorch,你可以按照以下步骤进行操作: 1. 首先,确保你的显卡支持 CUDA 10.2,并且已经安装了相应的驱动程序。 2. 下载安装 CUDA Toolkit 10.2,可以从 NVIDIA 官方网站下载对应版本的 CUDA Toolkit 安装程序。 3. 安装完成后,将 CUDA 的路径添加到环境变量中。在 Windows 系统中,你可以右键点击“我的电脑”,选择“属性”,然后点击“高级系统设置”,在弹出窗口中点击“环境变量”,找到“系统变量”中的“Path”变量,点击“编辑”,在变量值的最后加上 CUDA安装路径(例如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin)。 4. 安装 Anaconda(如果未安装)。你可以从 Anaconda 官方网站下载适合你系统的安装程序,并按照指示进行安装。 5. 打开 Anaconda Prompt(或命令行界面),创建一个新的 conda 环境并激活它: ``` conda create -n myenv python=3.7 conda activate myenv ``` 6. 在激活的环境中,使用以下命令安装 PyTorch: ``` conda install pytorch torchvision cudatoolkit=10.2 -c pytorch ``` 这将会安装适用于 CUDA 10.2PyTorch 和 torchvision。 7. 安装完成后,你可以在 Python 中导入 PyTorch 来验证安装是否成功: ```python import torch print(torch.__version__) ``` 如果没有出现错误信息,并且能够正确显示 PyTorch 的版本号,则说明安装成功。 希望这些步骤对你有帮助!如有其他问题,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值