Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn)

Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn)

安装Anaconda

pytorch-gpu安装

1:打开Anaconda prompt 创建虚拟环境

conda create -n pytorch python=3.6

2:创建后激活 环境

activate pytorch

3:到pytorch官网,根据安装的环境、python和CUDA版本查看指令
Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn)_第1张图片

输入

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

即可配置pytorch

4:安装完成后,输入下属指令进行测试

import torch
torch.cuda.is_available()

Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn)_第2张图片具体测试结果如上图所示,True显示pytorch-gpu安装成功。

你可能感兴趣的:(Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn))