安装pytorch之前需要先安装python和anaconda 如果没有安装,可以看我之前的两篇博客进行安装 CSDN和Anaconda安装,跟着安准成功_灵灵灵幺的博客-CSDN博客
安装完python和anaconda之后我们进入正题
首先我们需要创建一个虚拟环境,因为有时我们需要用到不同版本的pytorch,但我们肯定不能把pytorch卸载重新安装,这样太麻烦了,所以虚拟环境的创建很重要。
打开anaconda命令行输入
conda create -n pytorch python=3.11
这里的python是虚拟环境名 3.11是python版本号,这个根据自己的python版本进行填写
之后切换到这个虚拟环境
输入这行代码
conda activate pytorch
在浏览器输入https://pytorch.org/进入到pytorch官网,下滑找到这个界面
根据自己的情况选择,之后复制下面的代码,回到刚才创建的虚拟环境粘贴运行
运行结束后输入python进入python环境,输入
import torch
验证是否安装成功,不报错说明安装成功
输入
torch.cuda.is_available()
检验GPU是否可用,若可用则会返回True
好了,这就是pytorch安装的全部流程了,是不是很简单,现在网上很多教程都在误导,很麻烦,其实pytorch的安装并没有那么麻烦。