赞
踩
分享个人在安装pytorch中遇到的一些问题及解决办法。
为了使用显卡加速,需要安装双系统,因为虚拟机不支持显卡。安装推荐使用双硬盘。
如果配置环境出现难以定位或修改的问题,需要重装双系统里ubantu系统的,参考:
win10+ubuntu16.04双系统下完全删除并重装ubuntu16.04
之前没有安装过双系统的直接跳到此教程的第3步。
需要注意的几点:
由于ubantu自带nvdia驱动,但又不是完整,需要重新安装,参考:
Ubuntu18.04下安装Pytorch-GPU(超详细自己安装全过程)
需要注意的几点:
pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
安装后出现Please make sure that
source ~/.bashrc
CUDA 版本
cat /usr/local/cuda/version.txt
cuDNN 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
新版本查看cudnn命令已经更改,NVIDIA把版本信息换到cudnn_version.h中了,因此使用以下命令查看cudnn版本信息:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2没反应解决办法
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。