Windows11 pytorch安装教程

Windows11 pytorch安装教程,第1张

1. CUDA11.6

2. Anaconda4.10.3

3. 安装torch,在Anaconda prompt中依次输入以下命令:

conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

4. 验证是否安装成功

pip show torch

import torch
torch.cuda.is_available()

综上

CUDA 版本:11.6

Anaconda版本:4.10.3

pytorch版本:1.10.2+cu113

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/868034.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-13
下一篇2022-05-13

发表评论

登录后才能评论

评论列表(0条)

    保存