window 安装 PyTorch

window下 人工智能 Keras、TensorFlow、PyTorch、CUDA、cuDNN 的环境安装 总文章

window 安装 PyTorch 

人工智能AI：Keras PyTorch MXNet 深度学习实战（不定时更新）

---

1.安装链接：https://pytorch.org/get-started/locally/
  window下装的软件版本：CUDA 9.0、cuDNN 7.1.4、Python 3.6.5
  conda 在线安装命令：conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
  pip3 在线安装命令：pip3 install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
	            pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

2.测试一下是否安装好
		1.输入 import torch
		  输出 Traceback (most recent call last):
		  	File "", line 1, in 
				import torch
		  	File "G:\Anaconda3\lib\site-packages\torch\__init__.py", line 79, in 
				from torch._C import *
			ModuleNotFoundError: No module named 'numpy.core._multiarray_umath'
		  解决：查看numpy的当前版本 conda list numpy 或 pip show numpy
		        更新numpy的版本 pip install --upgrade numpy 或 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade numpy

		2.更新numpy的版本报错：Consider using the `--user` option or check the permissions.
		  解决：pip install --user --upgrade  numpy 

3.查看pytorch版本
	import torch
	print(torch.__version__)
	1.1.0

4.看看能不能用GPU加速：
	import torch
	torch.__version__
	Out[3]: '1.1.0'
	torch.cuda.is_available()
	Out[4]: True