环境配置--解决torch.cuda.is_available()返回:False

在网上很多都说是因为pytorch版本和CUDA不匹配造成的,但对于我的情况并不是,而是因为在官网默认下载到了的pytorch是cpu版本(真坑!另外在pycharm默认下载到的pytorch == 1.9.0也是cpu版本,用不了cuda)

1. 首先查看CUDA是否安装成功:

在终端输入:

ncvv --version

 

 有CUDA版本返回,证明是安装成功的;

2.查看pytorch是否安装成功

在终端输入python进入python环境:

python

 然后import 一下,并返回当前的版本号:

import torch

torch.__version__

 

证明pytorch也安装成功了,并确认cuda版本和pytorch兼容!(但是不知道这个torch是cpu版本还是cuda版本,在pycharm输入print(torch.__version__),如果返回结果后缀有-cpu那证明是版本错误) 

3.更换pytorch版本:

进入链接:https://download.pytorch.org/whl/torch_stable.html

直接点击下载CU100+ + win/linux(cu表示cuda版本,且版本号为10.0,以此类推)

下载完成后,在终端使用指令进行安装pytorch (后缀名也不能少!):

pip install cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值