怎么判断pytorch GPU版是否安装成功

 时间:2026-02-14 23:20:47

1、PyTorch由Torch7团队开发,以python为开发语言的深度学习框架,能实现GPU加速和动态神经网络

2、创建虚拟环境并镜像安装PyTorch

切换到虚拟环境肥躲,在dos命令行中执行conda activate torch2,如下图

看到torch2),说明选英光已经切换成功。

怎么判断pytorch GPU版是否安装成功

3、更换镜像源,按顺序运行番良以下命令

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 --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

4、运行执行安装Pytorch

运行:conda install pytorch torchvision cudatoolkit=10.1 成功安装


cuda这里选择的是10.1

5、在base虚拟环境下输入以下命令:

1、安装nb_conda:

conda install nb_conda

2、安装ipykernel

conda install ipykernel

3、为虚拟环境创建kernel文件:

conda install -n torch2 ipykernel

6、出现了Python[conda env:torch2]


在创建新的python文件时选中Python[conda env:torch2]即可。

怎么判断pytorch GPU版是否安装成功

7、然后我们在Jupyter Notebook测试PyTorch GPU版本安装是否成功

运行以下命令

import torch

x = torch.rand(5,5)

print(x)

输出类似下面内容,则pytorch安装成功

怎么判断pytorch GPU版是否安装成功

8、再执行

torch..cuda.is_available()

如果返回True,GPU版Pytorch成功安装完毕

  • 怎么看windows版本
  • 电脑怎么查看GPU信息
  • Win10系统资源管理器关闭了怎么重新打开
  • snipaste怎么使用?
  • 访问网页提示“你的连接不是专用连接”怎么办?
  • 热门搜索
    针灸减肥的利弊 红烧罗非鱼 睾丸疼痛怎么回事 顺辉瓷砖质量怎么样 网速不稳定怎么解决 职业装搭配 奇迹暖暖攻略 智能水表怎么偷水 对不起的英语怎么写 怎样减肥