官网打开可能较慢,耐心等待,可以正常打开。
链接:pytorch官网
2023年12月10日更新,演示版本2.1.1
如果中间有任何报错,参考最后一节的处理
一、说明和前期准备
1.pytorch是一个和tensorflow类似的框架
如果需要安装tensorflow,可以参考:
tensorflow 1,2 cpu+gpu(windows+linux)安装
2.安装anaconda
主要是为了一台机器安装多个互不干扰的环境。
安装最新版即可
anaconda 安装(windows+linux)
注意:安装anaconda要启动notebook(安装上面的教程来)
3.安装cuda
安装cuda11.8版本或12.1版本(rtx40系列装12.1版本容易一些)
如果你还有其他组件,应该检查互相之间的兼容性
链接:cuda windows安装
链接:cuda linux安装
链接:cudnn安装 win+linux
4.显卡(可选)
gpu(显卡)的在多数情况下的速度是超越cpu的,pytorch友好的一点是支持amd的ROCm
这样N卡和A卡都可以用(ROCm和cuda很多函数除了名字不一样基本都兼容)
所有架构都是cpu的坑少,初学通常用cpu
amd的ROCm目前linux是兼容最好,windows下只有少部分高端显卡支持,而且坑可能较多,所以现在建议还是用N卡,或者用linux+A卡
5.打开官网
linux用命令行,windows用cmd(也是命令行)
6.关于cpu和gpu版本的选择
建议装2个版本,因为通常cpu最稳,notebook可以很方便的切换内核
windows和linux都可以,不是说linux一定好。wsl2这个windows的linux子系统,个人用过几次,感觉还是坑多,如果windows下,用原生的就行。
从安装难易程度来说windows+cpu版本最容易入手。
二、安装pytorch(linux安装)
python版本为3.8-3.11,我这选用3.10
(一)linux+cpu
1.创建conda环境
conda create -n pytorch_cpu python=3.10
conda activate pytorch_cpu
2.使用conda安装
如果速度慢,可以设置conda为国内源(几个月前直连速度是很快的)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
3.搭配notebook
以下操作后才能在notebook下使用,否则没有这个环境文章来源:https://www.toymoban.com/news/detail-708451.html
pip install ipykernel -i https://pypi.tuna.tsinghua.edu.cn/simple/
python -m ipykernel install --name pytorch_cpu
4.进入notebook测试(从web进入)
import torch
print(torch.__version__)
显示结果如下
(二)linux+gpu(N卡)
1.创建conda环境
conda create -n pytorch_gpu python=3.10
conda activate pytorch_gpu
2.使用pip安装
注意:这里需要找对应的cuda版本
如果先装了cpu版本,用conda继续装,可能还是cpu版本,所以这里用pip安装
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
下载内容比较多,耐心等待,如果安装错误,再次运行命令即可
3.搭配notebook
以下操作后才能在notebook下使用,否则没有这个环境
pip install ipykernel -i https://pypi.tuna.tsinghua.edu.cn/simple/
python -m ipykernel install --name pytorch_gpu
4.进入notebook测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.backends.cudnn.enabled)
(三)linux+gpu(A卡)
和N卡安装方式一致,除了选ROMc,这里暂时不再额外写
手上没有独显a卡,无法测试。(如果使用rocm,建议在linux环境下开发,windows下坑一定很多)
三、安装pytorch(windows安装)
建议修改conda为国内源
windows的安装在cmd命令行中操作,或者在powershell命令行,就像这样:
(一)windows+cpu
1.创建conda环境
conda create -n pytorch_cpu python=3.10
conda activate pytorch_cpu
2.使用pip安装
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
3.搭配notebook
以下操作后才能在notebook下使用,否则没有这个环境
pip install ipykernel -i https://pypi.tuna.tsinghua.edu.cn/simple/
python -m ipykernel install --name pytorch_cpu
4.进入notebook测试
import torch
print(torch.__version__)
显示结果如下
(二)windows+gpu(N卡)
windows下不能用A卡的ROCm,据说ROCm正在努力兼容windows和其它平台,也许未来就可以用了
1.创建conda环境
conda create -n pytorch_gpu python=3.10
conda activate pytorch_gpu
2.使用conda安装
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
下载内容比较多,耐心等待,如果安装错误,再次运行命令即可
3.搭配notebook
以下操作后才能在notebook下使用,否则没有这个环境
pip install ipykernel -i https://pypi.tuna.tsinghua.edu.cn/simple/
python -m ipykernel install --name pytorch_gpu
4.进入notebook测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())
文章来源地址https://www.toymoban.com/news/detail-708451.html
到了这里,关于pytorch2.0.1 安装部署(cpu+gpu) linux+windows的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!