pytorch安装

• 1.Anaconda安装
• 2.创建环境
• 3.安装pytorch

1.Anaconda安装

下载链接
直接安装，能选择添加系统路径就添加。

不行的话，手动添加系统环境变量：

2.创建环境

启动Anaconda Prompt 创建虚拟环境：

创建名为 pytorch 的环境，代码：

conda create -n pytorch python=3.8

创建完成后，查看是否成功：

conda info --envs

3.安装pytorch

激活 pytorch 环境：

conda activate pytorch

官网选择对应版本的安装代码：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

注意cuda版本需要与电脑显卡匹配。可用nvidia-smi 查看电脑cuda版本。
为了加速可以更改镜像下载路径，更改C:\Users\用户名路径下的.condarc文件：

channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/- 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 custom_channels:conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmsys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudbioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmenpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudpytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudsimpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

安装完成即可。

测试是否成功：(在pytorch环境内)

python import torch torch.cuda.is_available

总结

以上是生活随笔为你收集整理的pytorch一直安装失败？的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。