13、pip3安装pytorch


pip3安装pytorch

1、进入官网下载相应的pytorch

2、离线下载轮子

可以直接复制Run this Command中的命令进行安装,但由于安装速度实在太慢,更换了源亦是如此。所以,为了加快安装速度,事先下载wheel文件,进行离线安装。

 浏览器打开上图红框网站

 使用迅雷下载,对应的版本。

3、使用pip3安装whl文件

 4、安装pytorch

由于已经安装好whl文件,所以再次使用pip3命令安装将非常快。

pip3 install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

 注意,在此步骤中会安装一些numpy、feature等packet,会影响一些安装速度,甚至出现一些错误。所以,可以选择提前安装这些packet。(提示缺什么安装什么)

5、测试

import torch

if __name__ == '__main__':
    print(torch.__version__)
    print(torch.cuda.is_available())
1.7.1
True

Process finished with exit code 0


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM