学习NLP需要使用pytorch 框架,由于电脑NVIDIA是8.0版本,而官网首页如下图
所以一时没找到,直接用conda 或pip方法,都有各种报错,折腾了很久。
事实上官网有cuda8.0版本,previous version.
我是windows,python36.
然后使用pip install 下载的torch包文件路径及文件名,就可以成功安装了。可能需要其他的包,比如numpy,但由于我是anaconda的base环境的python解释器,很多包都自动安装了,所以不大清楚。
安装完成后,进入python,运行命令
import torch
不报错就说明没问题啦。
检查是否能使用GPU
torch.cuda.is_available()
结果为TRUE。