pytorch环境
创建环境
conda create -n pytorch python=3.8
执行如下命令,激活这个环境。conda activate 虚拟环境名称
conda activate pytorch
Pytorch环境安装
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
测试安装完成
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
paddle环境
创建环境
conda create -n paddle python=3.8
执行如下命令,激活这个环境。conda activate 虚拟环境名称
conda activate paddle
paddlepaddle环境安装
conda install paddlepaddle-gpu==2.6.1 cudatoolkit=11.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge
测试安装完成
import paddle
print(paddle.utils.run_check())