window 安装 PyTorch

日萌社

人工智能AI:Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战(不定时更新)


Ubuntu 安装 CUDA、cuDNN、anaconda、pycharm、tensorflow、pytorch、JDK1.8

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

window 安装 PyTorch 

人工智能AI:Keras PyTorch MXNet 深度学习实战(不定时更新)


Ubuntu 安装 CUDA、cuDNN、anaconda、pycharm、tensorflow、pytorch、JDK1.8

PyTorch 安装 GPU版本(CUDA/cuDNN)

PyTorch 安装 GPU版本(CUDA/cuDNN)

window 安装 PyTorch


pytorch gpu 安装
	1.官网下载安装:https://pytorch.org/get-started/locally/
	  安装命令:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
	2.国内源直接下载链接
		网站链接:
                    window:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
                    linux:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
		选择下载:
                    window:pytorch-1.5.0-py3.7_cpu_0.tar.bz2
                    linux:pytorch-1.5.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
		下载链接:
                    window:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.5.0-py3.7_cpu_0.tar.bz2
                    linux:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.5.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
		安装:
                    window:conda install pytorch-1.5.0-py3.7_cpu_0.tar.bz2
                    linux:conda install pytorch-1.5.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2   
		pip list
			torch  1.5.0
	3.测试GPU版本的代码
		import torch
		#测试 CUDA
		print("Support CUDA ?: ", torch.cuda.is_available()) #返回True
		x = torch.tensor([10.0])
		x = x.cuda()
		print(x)
		y = torch.randn(2, 3)
		y = y.cuda()
		print(y)
		z = x + y
		print(z)
		# 测试 CUDNN
		from torch.backends import cudnn
		print("Support cudnn ?: ",cudnn.is_acceptable(x)) #返回True


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 "<ipython-input-79-eb42ca6e4af3>", line 1, in <module>
				import torch
		  	File "G:\Anaconda3\lib\site-packages\torch\__init__.py", line 79, in <module>
				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

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

あずにゃん

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值