在conda中安裝pytorch

• 一、創(chuàng)建conda環(huán)境
• 二、進(jìn)入環(huán)境
• 三、安裝pytorch
• 四、測試

一、創(chuàng)建conda環(huán)境

• 打開cmd，使用以下命令可以創(chuàng)建一個名為pytorch的環(huán)境，python版本是3.8。

conda create --name=pytorch python=3.8

二、進(jìn)入環(huán)境

• 剛剛創(chuàng)建的環(huán)境叫pytorch，所以用以下命令進(jìn)入這個環(huán)境：

activate pytorch

• 如果在命令行行首出現(xiàn)（pytorch）字樣，就是成功進(jìn)入環(huán)境了。如下圖：
  

• 如果沒有出現(xiàn)，那么可以通過anaconda打開這個環(huán)境：
  
  

三、安裝pytorch

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

• 目前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可以去官網(wǎng)了解。

四、測試

• 在這個環(huán)境下，在命令行輸入：

python

• 然后輸入：

import torch import torchvision torch.cuda.is_available()

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。