windows的pytorch安装

最新推荐文章于 2024-10-08 12:53:15 发布

是一只小咪罗呀

最新推荐文章于 2024-10-08 12:53:15 发布

阅读量200

点赞数

本文链接：https://blog.csdn.net/m0_49797510/article/details/110482115

版权

本文档介绍了在Windows上安装PyTorch的两种方法，包括通过pytorch官网的conda或pip命令，以及利用清华源进行快速下载安装。特别提示了在安装过程中可能遇到的问题及解决办法，如使用清华源加速下载，遇到下载停滞时按ESC键继续，以及确保安装cudatoolkit以避免import torch时出错。

摘要由CSDN通过智能技术生成

由于之前安装anaconda的时候已经安装了cuda（大概），而且cuda和GPU有关，所以这里暂时没提到关于cuda的安装。

这个文档可以解决import torch报错的问题

最基础的方法

pytorch官网：https://pytorch.org/

选择合适的conda或者pip方法，在dos窗口里复制粘贴即可。

（备注一个用清华源的conda下载方式，参考：https://www.cnblogs.com/yufeng97/p/13069247.html）

# 添加清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 添加pytorch镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes

然后复制之前的conda代码