Pytorch安装的一些问题

本文介绍了在Windows上安装PyTorch时遇到GPU不可用的问题及解决过程。尝试更新CUDA、GPU驱动、下载CuDNN,但问题仍未解决。发现1050ti显卡未在官方支持列表,且通过conda安装的PyTorch可能是CPU版本。通过查询所有可用版本并手动指定GPU版本进行安装,最终解决问题。
摘要由CSDN通过智能技术生成

Pytorch(conda)安装

Pytorch安装(conda)的一些问题

问题:
最近安装了一下pytorch,前面安装了tensorflow没有遇到什么问题,就是直接在pycharm中搜索tensorflow-gpu版本并安装,gpu可用。但是照猫画虎的安装pytorch之后torch可用,但是Gpu不可用,torch.cuda.is_available()返回False。

尝试解决方法
1.更新Cuda,从此处下载cuda版本:下载地址参考文章:参考。一般情况参考中遇到的问题你都要解决,包括visual studio安装失败,nsight xxx安装失败等。安装目录最好不建议默认(刚开始没敢改结果少了几G)
仍存在问题:安装了10.1版本参考他人不同查看cuda方法看淡方法不同:

nvcc -V
结果:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243
nvidia-smi
结果:
NVIDIA-SMI 442.59       Driver Version: 442.59       CUDA Version: 10.2

最终并没有处理这个问题,(可能这两者反应的不是同一个问题,根本没有关联?)。
2. 更新Gpu 驱动,此处就不添加参考了。
3. 下载cudnn,下载地址:下载地址,将压缩包内相应的文件放入文件夹:https://developer.nvidia.com/cudnn下相应文件夹内覆盖。

最终上述的方法并没有解决gpu不可用的问题。
查看gpu是否支持:官网,最终发现本人1050ti没有在支持列表内,但是(tf可用?)这也在安装cuda的时候有体现,会warning,最终确认可以用。

常用方法:完成上述的gpu准备后,在官网选择自己自己相应的版本命令安装,(去掉“-c pytorch”命令参数,不然使用官网源速度慢)。但是我使用上述方法gpu仍不可用。

怀疑使用的torch是cpu版本(但是自己确实使用的非cpu版本安装命令),这个部分博客也说了这个问题,window下官网方法下载的是cpu版本,于时使用conda remove卸载原始的安装包,发现确实是cpu版本的。
卸载详情:

## Package Plan ##

  environment location: E:\anaconda

  removed specs:
    - pytorch


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    intel-openmp-2020.0        |              166         1.5 MB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    mkl-2020.0                 |              166        98.9 MB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    ------------------------------------------------------------
                                           Total:       100.5 MB

The following packages will be REMOVED:

  _pytorch_select-1.1.0-cpu
  ninja-1.7.2-0
  pytorch-1.0.1-cpu_py37h39a92a0_0
  torchvision-0.2.1-py_2

在此过程中更改了conda channel,再使用官网方法安装报错。

Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Collecting package metadata (repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.

PackagesNotFoundError: The following packages are not available from current channels:

  - torchvision

Current channels:

  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
  - https://mirrors.tuna.tsinghua.edu
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值