在conda中安裝pytorch

• 一、創建conda環境
• 二、進入環境
• 三、安裝pytorch
• 四、測試

一、創建conda環境

• 打開cmd，使用以下命令可以創建一個名為pytorch的環境，python版本是3.8。

conda create --name=pytorch python=3.8

二、進入環境

• 剛剛創建的環境叫pytorch，所以用以下命令進入這個環境：

activate pytorch

• 如果在命令行行首出現（pytorch）字樣，就是成功進入環境了。如下圖：
  

• 如果沒有出現，那么可以通過anaconda打開這個環境：
  
  

三、安裝pytorch

• 在pytorch官網（https://pytorch.org/get-started/previous-versions/）可以查看不同pytorch版本在不同操作系統的各種安裝方式。

• 目前pytorch最新版本是1.8.0的。

• 我的顯卡的CUDA版本是11，所以這里我使用：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

• 如果顯卡不支持CUDA，也可以安裝CPU版本的pytorch，例如使用：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch

• 其他版本的pytorch可以去官網了解。

四、測試

• 在這個環境下，在命令行輸入：

python

• 然后輸入：

import torch import torchvision torch.cuda.is_available()

• 如果像下圖一樣沒有報錯，就是安裝成功了。
  

總結

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

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