pytorch
学习需要安装pytorch,记录一下自己的安装过程!
查看Python版本
先查看CUDA版本
nvidia-smi
在线安装进入官网下载
选择对应的版本和安装方式!
但是下载速度慢,所以可以离线安装
离线下载安装
pytorch与torchvision下载链接:
https://download.pytorch.org/whl/torch_stable.html
CUDA没有11.6的我选择11.5,Python版本3.8,Windows64位
下载完pytorch和torchvision的.whl文件之后,在下载的目录内使用下列命令即可:
pip install torch-1.11.0+cu115-cp38-cp38-win_amd64.whl
pip install torchaudio-0.11.0+cu115-cp38-cp38-win_amd64.whl
是否安装成功
运行代码看能否使用
import torch
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/147428.html