pytorch安裝

• 1.Anaconda安裝
• 2.創建環境
• 3.安裝pytorch

1.Anaconda安裝

下載鏈接
直接安裝，能選擇添加系統路徑就添加。

不行的話，手動添加系統環境變量：

2.創建環境

啟動Anaconda Prompt 創建虛擬環境：

創建名為 pytorch 的環境，代碼：

conda create -n pytorch python=3.8

創建完成后，查看是否成功：

conda info --envs

3.安裝pytorch

激活 pytorch 環境：

conda activate pytorch

官網選擇對應版本的安裝代碼：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

注意cuda版本需要與電腦顯卡匹配。可用nvidia-smi 查看電腦cuda版本。
為了加速可以更改鏡像下載路徑，更改C:\Users\用戶名路徑下的.condarc文件：

channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/- defaults show_channel_urls: true channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda default_channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 custom_channels:conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmsys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudbioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmenpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudpytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudsimpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

安裝完成即可。

測試是否成功：(在pytorch環境內)

python import torch torch.cuda.is_available

總結

以上是生活随笔為你收集整理的pytorch一直安装失败？的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。