anaconda + pytorch 配置 CPU版和GPU版

配置国内源

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/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes

# 删除所有channels
conda remove-key channels
# 删除指定channels
conda remove channels $name

创建环境

conda create -n pytorch python=3.7

进入pytorch环境

conda activate pytorch

自动配置

GPU

在porch官网，选择配置。
注意1：（GPU）有Nvidia的显卡才能选择CUDA，
注意2：使用国内镜像要省略命令 -c pytorch -c nvidia

conda install pytorch torchvision torchaudio pytorch-cuda=11.6

或者（巨慢）

# 指定一个pytorch版本
conda install pytorch=1.12.0

CUP

# cpu最新版本
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

指定版本下载

pip install torch==版本号 torchvision==版本号 torchaudio==版本号 -i https://pypi.tuna.tsinghua.edu.cn/simple

手动配置GPU

在cuda的文档中下在对应版本的cuda+torch+torch+torchvision+torchaudio
例如：cuda11.6版的torch1.12.0、python3.7、windows，与之相匹配的是
cuda11.6版的torchaudio0.12.0、python3.7、windows，
cuda11.6版的torchvision0.13.0、python3.7、windows

手动安装

# 进入pytorch环境打开文件下载目录
(pytorch) PS C:\Users\name> cd 'C:\Download\'
# 安装 cu116 和 torch-1.12.0
(pytorch) PS C:\Download> pip install torch==1.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
(pytorch) PS C:\Download> pip install .\torch-1.12.0+cu116-cp37-cp37m-win_amd64.whl
# 安装torchaudio和torchvision
(pytorch) PS C:\Download> pip install .\torchaudio-0.12.0+cu116-cp37-cp37m-win_amd64.whl
(pytorch) PS C:\Download> pip install .\torchvision-0.13.0+cu116-cp37-cp37m-win_amd64.whl

测试

GPU

CUP

anaconda + pytorch 配置 CPU版和GPU版相关推荐

win10+anaconda+cuda配置dlib，使用GPU对dlib的深度学习算法进行加速（以人脸检测为例）...
win10+anaconda+cuda配置dlib,使用GPU对dlib的深度学习算法进行加速(以人脸检测为例) 转载于:https://www.cnblogs.com/zhehan54/p/8540 ...
Windows 10 Anaconda Python 3.7 安装 MXNet GPU版
MXNet CPU版本可以直接通过pip进行安装: pip install mxnet MXNet GPU版本不推荐通过pip进行安装,推荐使用conda进行安装: conda install -c ...
ubuntu 16.04 Anaconda3 中安装tensorflow环境[CPU版和GPU版]
首先,安装Anaconda3, 安装教程:https://blog.csdn.net/zdx1996/article/details/88117014 在anaconda环境下,先确认要在anacon ...
全网最详细之如何安装gpu版的pytorch
安装gpu版pytorch 一.准备: 1.需要查看nvidia驱动是否安装安装驱动可参考:https://blog.csdn.net/qq_41428418/article/details/119 ...
C++ | PaddleOCR GPU版使用步骤与测试时间对比(相对CPU)
点击下方卡片,关注"OpenCV与AI深度学习"公众号视觉/图像重磅干货,第一时间送达! 导读本文主要介绍C++版PaddleOCR GPU版的使用步骤和测试时间对比(相对CP ...
gpu版pytorch安装
pytorch pytorch官网:https://pytorch.org/get-started/locally/ 目前gpu版pytorch安装不需要自己安装cuda,而且注意windows版仅支 ...
OpenCV中GPU版MOG2的使用
最近项目中的GMM算法,需要改成GPU版,以降低CPU占用.重新编译了opencv2.4.13,网上很多教程,不再赘述.说一下自己遇到的坑.项目用caffee,cuda版本8.0,opencv2.4. ...
深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】
深度学习环境配置入门深度学习,首先要做的事情就是要搭建深度学习的环境.不管你是Windows用户,Mac用户还是Ubuntu用户,只要电脑配置允许,都可以做深度学习,毕竟Windows.Mac和Ub ...
pytorch无坑超详细图文CPU版小白安装教程(配gpu版链接、conda命令教程)
想安装gpu版本的朋友们请移步gpu版pytorchan安装教程直达文章目录创建.激活.退出.删除环境法一:官网默认指令安装(可能比较慢) 法二:更换清华源下载法三:下载包安装版本对应问题 ...