windows清华镜像pip三步安装pytorch

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:

windows清华镜像pip三步安装pytorch_第1张图片

如果要安装GPU版本的pytorch,就需要知道计算机的cuda版本,可以通过以下两种方法查看cuda版本:

        方法1:在命令行输入“nvcc -V”,但当计算机没有安装对应程序时会报错如下

        这时候就可以使用方法2。 

         方法2:在命令行输入“nvidia-smi”查看显卡占用情况

windows清华镜像pip三步安装pytorch_第2张图片

 在返回结果的右上角可以查看到cuda版本。

 

Step3:

使用“win+R”,输入“cmd”打开命令行工具,将Step2中复制的command粘贴进去并运行即可。windows下命令行的粘贴快捷键是“shift+ins”。

下图即为安装过程,当出现“Successfully installed....”即表示pytorch安装成功。

windows清华镜像pip三步安装pytorch_第3张图片

测试安装结果:

在命令行下输入python进入python程序,导入pytorch并打印版本信息:

windows清华镜像pip三步安装pytorch_第4张图片

你可能感兴趣的:(深度学习,python,pytorch)