欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

pytorch的环境配置

时间:2023-08-22

参考:https://blog.csdn.net/weixin_44791964/article/details/106037141

我的pytorch环境配置需求:

python 3.6.5pytorch 1.9.0 cuda 10.2numpy 1.19.2matplotlib 3.1.0

1.Anaconda的下载和安装 (参考上面链接) 2.Cudnn和CUDA的下载和安装(参考上面链接)

CUDA(百度网盘有cuda10.2版本,可以直接下载使用。)

3.配置torch环境

第一步:

conda create –n pytorch python=3.6.5

第二步:

activate pytorch

第三步:下载gpu版本的pytorch

pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html

第四步:再安装其他需要的包,比如numpy等

最后

在pycharm中运行测试是否成功

import torchprint(torch.__version__) //pytorch是cpu还是gpu版本print( torch.zeros(1).cuda())print(torch.cuda.is_available()) //电脑的 GPU 能否被 PyTorch 调用

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。