PyTorch环境搭建

显卡驱动下载安装

查看电脑显卡类型。
win10系统查看电脑显卡类型：开始界面选择设置-关于-设备管理器-显示适配器。

显卡驱动下载：https://www.nvidia.cn/Download/index.aspx?lang=cn
根据电脑显卡型号进行选择。

Anaconda下载

下载地址：https://www.anaconda.com/products/distribution#Downloads

下载后按照提示安装即可，安装完成后在开始界面Anaconda3文件夹：

PyTorch安装

官网地址：https://pytorch.org/get-started/locally/

根据需要选择参数，复制指令，在Anaconda Prompt窗口粘贴执行。


相关安装包加载完成后，出现Proceed([y]/n)？时输入y即进行安装，由于下载源问题下载安装速度会比较慢，或者安装失败，最好选择不限制的网络进行安装。（我开始在公司安装总是失败，查看很多帖子说是网速问题，但我试了公司网络访问国外网站就会很慢，后来在家里安装成功的，前前后后安装了好多次）

在安装完成之后，需要检查一下是否安装成功，可以直接查看当前环境下安装的所有包来验证，输入命令

conda list

创建虚拟环境

打开anaconda prompt或者cmd，输入命令

conda create -n pytorch python==3.9
其中“pytorch”为创建虚拟环境的名称，“3.9”为python的版本号。

执行完成后，输入命令

conda env list

进入新创建的虚拟环境，输入命令

activate pytorch

如下图：

删除虚拟环境

//先退出虚拟环境
conda deactivate
//查看环境列表
conda env list
//选择需要删除的虚拟环境，如果一个变量名有多个地址，加上地址进行删除
conda env remove -p d:\lele\pytorch学习\pytorch_simple_teaching_example-main\.conda
//如果删除对应所有地址，例子中的pytorch代表要删除的环境变量名称
conda remove -n  pytorch --all

执行如图：

添加环境变量

运行PyTorch

Anoconda Promp运行

先进入python

执行py文件

import os
os.system(’python demo.py’)

vscode运行PyTorch

安装python插件。



以管理员身份运行PowerShell后依次执行以下命令：

conda init powershell
Set-ExecutionPolicy RemoteSigned
get-ExecutionPolicy

命令执行完后再次打开vs code在终端中可以看见进入anaconda环境中，科执行canda命令，可切换到自己创建的环境中执行对应程序。

参考文章

1. PyTorch环境搭建
2. 检查pytorch是否安装成功、查看torch和cuda的版本
3. 在VSCode中使用Pytorch遇到的问题