win 10 +cuda9.2+pytorch0.4.1(gpu)安装

如果之前下载了别的torch或是pytorch最好先删除,不然后面安装torch总有奇奇怪怪的问题。
一、
cuda+cudnn先前已经安装好的。选择和显卡对应版本的cuda下载安装即可。
再选对应cudnn下载
打开压缩包,依次将压缩包中的3个文件复制到相应位置

  1. …\cuda\bin\cudnn64_7.dll —> …\NVIDIA GPUComputing
    Toolkit\CUDA\v9.1\bin
  2. …\cuda\include\cudnn.h —> …\NVIDIA GPUComputing
    Toolkit\CUDA\v9.1\include
  3. …\cuda\lib\x64\cudnn.lib —>…\NVIDIA GPUComputing
    Toolkit\CUDA\v9.1\lib\x64

二、pytorch
根据python版本和cuda版本去官网https://pytorch.org/下载对应whl文件.(这是我下载的版本win+python37+cuda9.2+torch0.4.1老版本的下载入口
win+python37+cuda9.2+torch0.4.1
在whl文件下载目录下cmd进入命令行,pip install安装(前面加的是国内镜像源)

pip install  -i https://pypi.doubanio.com/simple/  --trusted-host pypi.doubanio.com  torch-0.4.1-cp37-cp37m-win_amd64.whl

在这里插入图片描述
三、验证安装是否成功

 import torch
 print(torch.__version__)
 print(torch.version.cuda)
 print(torch.cuda.is_available())

在这里插入图片描述

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值