养生 装修 购物 美食 感冒 便秘 营销 加盟 小吃 火锅 管理 创业 搭配 减肥 培训 旅游

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

时间:2024-10-27 22:19:51

pythorchGPU版版安装成功检测

工具/原料

电脑

方法/步骤

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

2、创建虚拟环境并镜像安装PyTorch切换到虚拟环境,在dos命令行中执行condaactivatetorch2,如下图看到torch2),说明已经切换成功。

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

3、更换镜像源水貔藻疽,按顺序运行以下命令condaconfig--addchannelshttps://mirrors.tu荏鱿胫协na.tsinghua.edu.cn/anaconda/pkgs/free/condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/condaconfig--setshow_channel_urlsyescondaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

4、运行执行安装Pytorch运行:condainstallpytorchtorchvisioncudatoolkit=10.1成功安装cuda这里选择的是10.1

5、在base虚拟环境下输入以下命令:安装nb_conda:condainsta盟敢势袂llnb_conda安装ipyke鸡堕樱陨rnelcondainstallipykernel为虚拟环境创建kernel文件:condainstall-ntorch2ipykernel

6、出现了Python[condaenv:torch2]在创建新的python文件时选中Python[condaenv:torch2]即可。

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

7、然后我们在JupyterNotebook测试PyTorchGPU版本安装是否成功运行以下命令importtorchx=torch.rand(5,5)print(x)输出类似下面内容,则pytorch安装成功

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

8、再执行torch..cuda.is_available()如果返回True,GPU版Pytorch成功安装完毕

© 一点知识