search

SINLDY / -linux-cellpose

0 stars 0 forks source link

无法调用GPU #1

Open SINLDY opened 3 months ago

SINLDY commented 3 months ago

由于我的显卡为AMD Radeon™ RX 6700 XT，我似乎无法使用PyTorch来调用我的GPU，并使用其来处理cellpose的大量图像

SINLDY commented 3 months ago

PyTorch似乎支持使用ROMs来调用AMD的GPU，但是仅限linux系统，我尝试安装该系统到我的电脑中

SINLDY commented 3 months ago

安装完毕，我需要使用 sudo apt update sudo apt install "linux-headers-$(uname -r)" "linux-modules-extra-$(uname -r)" sudo usermod -a -G render,video $LOGNAME # Add the current user to the render and video groups wget https://repo.radeon.com/amdgpu-install/6.2/ubuntu/noble/amdgpu-install_6.2.60200-1_all.deb sudo apt install ./amdgpu-install_6.2.60200-1_all.deb sudo apt update sudo apt install amdgpu-dkms rocm 指令来在Linux上安装ROCm，以此调用GPU

SINLDY commented 3 months ago

使用指令：、 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1 安装PyTorch

SINLDY commented 3 months ago

截图 2024-08-11 13-58-32 直接吃了我35GB，还好我安装系统的时候给了200GB的容量，看来之后得买新硬盘了

SINLDY commented 3 months ago

还是不行，我重启一下试试