windows安装pytorch环境
由于CUDA Version更新到11.7了,因此,本教程也同步更新
安装Anaconda
(1)首先打开Anaconda官网,下载对应平台的安装包
Anaconda官网
我们这里安装的包是Anaconda3-2022.10-Windows-x86_64.exe
然后,双击exe文件,开始安装
等待安装结束(这里选择Just me即可)
注:这里有一种方法可以自动添加环境变量,在安装的过程中,勾选自动添加到Path选项即可!这样做可跳过下面的步骤
(2)安装完成后,点击系统属性设置——添加环境变量(安装时勾选自动添加到Path选项的可跳过此步骤):
在系统Path路径下添加(高亮部分的两行内容,具体的安装路径以机器上为准):
D:\software\anaconda3\Scripts
D:\software\anaconda3\Library\bin
(3)打开CMD命令,输入conda,能正常显示,说明已经安装成功了:
安装Pytorch
(1)初始化.condarc文件
conda config --set show_channel_urls yes
这个时候,我们打开C盘用户下,可以看到多了一个.condarc文件
(2)打开这个文件,按照以下的方式进行编辑:
- 先设置虚拟环境的根目录
envs_dirs:
- D:\software\anaconda3\envs
- 然后添加channels:
- 清华大学镜像(安装速度快,但有时最新版不一定能安装成功)
- 阿里云镜像(安装速度慢,但package内容较全)
channels:
- defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
- http://mirrors.aliyun.com/anaconda/pkgs/main
- http://mirrors.aliyun.com/anaconda/pkgs/r
- http://mirrors.aliyun.com/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
(3)创建pytorch的虚拟python环境:
(打开cmd命令行窗口,prefix后添加指定安装路径)
conda create -n pytorch python=3.9
输入y,开始创建虚拟环境
创建完成后,激活虚拟环境,并安装pytorch:
conda activate pytorch
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
等待安装完成后,即可使用pytorch环境了
(4)检测pytorch环境是否安装成功
import torch
print(torch.cuda.is_available())
True
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· .NET10 - 预览版1新功能体验(一)