安装pytorch模块
这里是安装pytorch的第二个步骤,正式进入了pytorch模块。
首先,如果你想要使用pytorch,那么你的电脑上面必须要有英伟达的显卡,那么怎么看自己电脑上面有没有英伟达的显卡呢?这里可以去看我的另一篇文章,查看自己电脑上面是否由英伟达的显卡。
https://blog.youkuaiyun.com/congcong0509/article/details/127753424?spm=1001.2014.3001.5501
第二步,安装pytorch模块
但是在安装pytorch模块之前,先需要确定一下自己电脑的配置。
查看自己电脑CUDA的版本,windows+R,输入cmd命令,打开终端命令窗口。
输入:nvidia-smi
这里可以看到,我的CUDA版本是12.0(这个版本我估计是最高了的),CUDA版本不是越高越好,因为CUDA越高,所需要安装的CUDA驱动的版本也好很高。
注意:CUDA安装的驱动版本要高于CUDA的版本。
通过官网命令进行pytorch的安装。下面介绍的就是通过官网命令进行安装的。
好了,下面进行pytorch的安装流程。
1、在安装完成Annconda之后,进入到Annconda Prompt应用里,点击打开。
2、进入之后,创建一个自己的虚拟环境,这个虚拟环