在Anaconda环境下pytorch安装(cuda10.0版本)
·
废话不多说,直接开始吧
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
成功了~
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)