最近在win10裝torch，發(fā)現(xiàn)沒有Linux那么便捷，用conda和pip裝默認版本都是cpu，而且下載速度很慢。網(wǎng)上的解決方法有好幾種，但是初學者可能面臨的問題是：我的系統(tǒng)該裝哪個版本的？cuda、torch、torchvision以及 torchaudio裝哪個版本？以及裝了cpu版本的顯示torch.cuda.is_available()  False 該怎么辦？

那么，方法如下（親測成功好用）

第一步：創(chuàng)建環(huán)境 （-n +環(huán)境名（torch) + python版本（3.10））

conda create -n torch python=3.8

第二步：了解你電腦的cuda 版本，在terminal中輸入：nvidia-smi 即可，如下圖所示，版本為12.3

第三步：去pytorch官網(wǎng)看你的conda適配的torch、torchvision以及 torchaudio版本，這個鏈接：

Previous PyTorch Versions | PyTorch

比你的CUDA低版本的都能安裝，看好對應的版本就可以了，不用官網(wǎng)的命令安裝（會很慢）

第四步：開始安裝，鑒于使用conda的各種問題，我們這里不使用conda安裝，使用pip安裝，根據(jù)剛才看好的版本，在創(chuàng)建的torch環(huán)境下輸入命令：

#進入torch環(huán)境
conda activate torch

#用鏡像源快速安裝對應版本
pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

第五步：驗證torch安裝是否成功，import torch成功說明安裝成功，如果torch.cuda.is_available()返回False 時，說明裝了cpu版本的。不要慌，往下看。

第五步：查看torch環(huán)境下所有庫：

conda list

可以看到，pytorch是cpu版本的

第六步：手動去清華源下載對應的gpu版pytorch，和上面圖上看到的cpu的版本保持一致即可

Index of /anaconda/cloud/pytorch/ | 清華大學開源軟件鏡像站 | Tsinghua Open Source Mirror

比如pytorch版本是1.10.1（需要對應），環(huán)境創(chuàng)建時python版本是3.8 （需要對應），cuda 12.3（比它小的都可以），那么下面幾個可以下載，選一個下載，最好裝到一個英文路徑下。

第七步：離線安裝剛剛下載的文件

#  cd 到下載的文件目錄，例如安裝目錄是E:\Download
cd E:\Download

#離線安裝
conda install --offline pytorch-1.10.0-py3.6_cuda10.2_cudnn7.6.5_0.tar.bz2

最后一步： conda list, 查看pytorch 是否已經(jīng)被替換。可以看到替換成GPU版本了

再輸入 torch.cuda.is_available() 驗證，返回True .大功告成！

總結(jié)

到此這篇關(guān)于快速安裝gpu版本的pytorch的文章就介紹到這了,更多相關(guān)快速安裝gpu版本pytorch內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論