这里写自定义目录标题
如何安装pytorch
我也是小白,只是在安装过程中出现了一些问题,记录下来可能以后会用到。
首先在pytorch官网选择自己要下载的版本。
官网:https://pytorch.org/
比如说我的是上面图片显示的。会有一个Command,然后打开Anaconda Prompt,复制粘贴上面的Conmand就会下载安装
问题
由于国内网下载很慢,而且有两个比较大的文献:cudatookit和pytorch的文件需要下载。然后我下载了一晚上,第二天提示没有安装上。(当时没有截图,主要是说HTTP导致下载失败),但它提示了有两个网址下载上面的两个文件。
如何解决的
将这两个网址使用迅雷下载就很快了。
如上图,可能几分钟就下载好了。
然后就是安装。
在Anaconda Prompt中输入conda install 路径\文件名和后缀。这两个都需要安装。
安装很快,完成。
之后就是验证pytorch安装好没有
输入代码验证:
import torch
cuda_gpu = torch.cuda.is_available()
if (cuda_gpu):
print("Great, you have a GPU!")
else:
print("Life is short -- consider a GPU!")
结果:
参考博客:https://blog.csdn.net/cuixing001/article/details/81952116