【测试pytorch是否可以使用】

判断pytorch-gpu版本是否可以使用

文章目录

  • 测试pytorch是否可以使用
  • 一、pycharm
  • 二、查看Anaconda-pytorch
    • 1.进入Anaconda prompt
    • 2.查看Anaconda环境
    • 3.结果
  • 总结


测试pytorch是否可以使用

`

工具:Anaconda+Pycharm


一、pycharm

核心代码:demo.py

import torch
import torchvision
print(torch.cuda.is_available())

这时候显示False,以下开始排查原因

二、查看Anaconda-pytorch

1.进入Anaconda prompt

代码如下(示例):


(base) C:\Users\Conan>activate pytorch

(pytorch) C:\Users\Conan>python
Python 3.6.8 |Anaconda, Inc.| (default, Feb 21 2019, 18:30:04) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>>

此时显示True,说明问题出现在pycharm

2.查看Anaconda环境

发现当初下载了好几个pytorch(配置的时候没有搞清楚哪个能用哪个不能用)----->发现原因:pycharm project interpreter设置错误。
【测试pytorch是否可以使用】_第1张图片

3.结果

True


总结

1.排除了anaconda是否没有正确安装
2.排查pycharm配置问题
3.最好不要有重名(迷惑性BUG)

你可能感兴趣的:(小白捉虫,pytorch,python,人工智能,深度学习)