如何安裝 PyTorch

2022-09-20 08:12:14 字數 837 閱讀 6373

現在使用 pytorch 框架進行深度學習的人越來越多了,但是 torch 的安裝(gpu 版本)和普通第三方包的安裝過程不太一樣,因為涉及到電腦的顯示卡驅動版本,下面來看看安裝 pytorch 的正確姿勢。

開啟 nvidia 控制面板,點選工具欄幫助 --> 系統資訊,在彈出的對話方塊中有如下資訊:

可以看到本機驅動程式版本為 466.92,接著前往 nvidia cuda toolkit release notes,在cuda driver這一節有下面這個表:

由於本機為 windows 系統且驅動版本為 466.92,所以應該選擇 cuda 11.6.x。

在 pytorch 官網的 get started 頁面有下圖所示的安裝嚮導:

雖然沒找到 cuda 11.6.x,但是 cuda 11.3 也是可以的,複製run this command裡面的指令:

pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f
這裡把 pip3 換成了 pip,不過 pip3 也是可以的。在終端執行上述指令,等待安裝完成即可。

如何安裝pytorch

離線安裝 用conda安裝會比較方便,下面直接用conda安裝 conda config add channels conda config add channels conda config add channels 或者,直接改.condarc檔案 注意 沒有自己電腦gpu,就不要安裝cuda了...

PyTorch學習 安裝PyTorch

例如,使用的是 windows 系統,想用 pip 安裝,python 是 3.6 版的,沒有 gpu 加速,那就按上面的選,然後根據上面的提示,在 terminal 中輸入以下指令就好了 pip3 install torch 1.3.1 cpu torchvision 0.4.2 cpu ftor...

如何安裝CPU版本的pytorch

二 配置工作環境過程中可能會遇到的一些安裝庫的命令 pip install numpy 安裝numpy庫 pip uninstall numpy 解除安裝numpy庫 conda install numpy conda uninstall numpy 同上 安裝gpu版本的pytorch只需要再安裝...