【测试pytorch是否可以使用】

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

---

测试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设置错误。

3.结果

True

---

总结

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