在anaconda创建并指定安装GPU版本pytorch环境如下:
- 创建环境
conda create -n torch38 python=3.8
- 查找conda下pytorch版本–关键步骤
conda search pytorch
这是官网安装pytorch的命令,但是输入后直接会给我安装成cpu版本(下图中蓝色框中版本)。实际上要安装的是红色框中版本!
# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
3. 安装pytorch
在pytorch中添加上具体的版本号,这样conda这个小机灵鬼就不会装错了!
conda install pytorch=1.7.1=py3.8_cuda11.0.221_cudnn8.0.5_0
conda install pytorch=1.12.1=py3.9_cuda11.3_cudnn8.3.2_0
conda install torchvision==0.8.2=py38_cu110
conda install torchvision==0.13.1=py39_cu113
conda install torchaudio==0.7.2=py38
conda install torchaudio==0.12.1=py39_cu113
- 测试
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
【参考文献】
conda自动装cpu版torch原理
安装gpu版本
conda install pytorch==1.10.1=py3.8_cuda11.3_cudnn8.2.0_0
conda install torchvision==0.11.2=py38_cu113
conda install torchaudio==0.10.1=py38_cu113
conda install tqdm
pip install opencv-python
conda install matplotlib