logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Anaconda安装虚拟cuda环境——解决Pytorch无法识别cuda的问题

在 Anaconda 的环境配置过程中,明明根据电脑已经安装的 cuda 版本从 Pytorch 官网上安装了对应版本的 torch 包 但依然无法调用 GPU 进行 torch 任务,编辑 GPUtest.py 文件。得到结果如下:torch 无法正常识别到 cuda。打开 Anaconda Prompt 输入,nvidia-smi 查看当前 GPU 驱动程序nvidia-smi得到驱动版本是5

文章图片
#pytorch#人工智能#python +1
到底了