win10下安装pytorch步骤

最新推荐文章于 2023-08-16 18:22:55 发布

原创最新推荐文章于 2023-08-16 18:22:55 发布 · 223 阅读

0 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

54 篇文章

订阅专栏

本文详细介绍了在本地环境中安装PyTorch的步骤，包括CUDA和cuDNN的安装，使用Anaconda进行环境配置，以及如何设置清华源加速依赖包的下载。确保CUDA版本与PyTorch兼容是成功安装的关键。

部署运行你感兴趣的模型镜像

步骤：
1）首先安装本机支持的cuda和cudnn，去官网下载
2）安装anaconda
3）更改conda库

  		  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
          conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
          conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
          conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
          conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
          conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
          conda config --add channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
          conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
          
 		   conda config --set show_channel_urls yes

4)更新cuda

cuda update cuda
或者
cuda update --all

5)安装pytorch

 conda install pytorch torchvision cudatoolkit=10.1

注意cudatoolkit是你的cuda版本，要对应好

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理