在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()
- 如果像下图一样没有报错,就是安装成功了。