废话不多说,直接开始吧

1.进入官网在这里插入图片描述

点击以往版本发现只有v1.2有cuda10选项
在这里插入图片描述

2. 打开Anaconda Prompt

新建pytorch环境

(base) C:\Users\RUAN>conda create -n pytorch python=3.6

建好之后启动

(base) C:\Users\RUAN>conda activate pytorch

安装pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

3.验证是否装好GPU

print torch.cuda.is_available()

输出

True

成功了~

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐