Step1:
1.1 在C盘的路径 “C:\Users\wang123\AppData\Roaming” 下创建名为pip的文件夹,然后再在该文件夹下创建名为“pip.ini”的文件。
有时候可能找不到AppData文件夹,这时候在文件夹导航栏中选择“查看-->显示/隐藏-->隐藏的项目”,勾选“隐藏的项目”即可显示出AppData文件夹。
1.2 复制下面代码进入pip.ini文件,保存并退出
[global]
index-url=https://pypi.tuna.tsinghua.edu.cn/simple/
[install]
trusted-host=pypi.tuna.tsinghua.edu.cn
Step2:
在pytorch官网选择合适的pytorch版本,复制对应的command:
如果要安装GPU版本的pytorch,就需要知道计算机的cuda版本,可以通过以下两种方法查看cuda版本:
方法1:在命令行输入“nvcc -V”,但当计算机没有安装对应程序时会报错如下
这时候就可以使用方法2。
方法2:在命令行输入“nvidia-smi”查看显卡占用情况
在返回结果的右上角可以查看到cuda版本。
Step3:
使用“win+R”,输入“cmd”打开命令行工具,将Step2中复制的command粘贴进去并运行即可。windows下命令行的粘贴快捷键是“shift+ins”。
下图即为安装过程,当出现“Successfully installed....”即表示pytorch安装成功。
测试安装结果:
在命令行下输入python进入python程序,导入pytorch并打印版本信息: