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版本查看指令

输入
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
即可配置pytorch
4:安装完成后,输入下属指令进行测试
import torch
torch.cuda.is_available()
具体测试结果如上图所示,True显示pytorch-gpu安装成功。
更多推荐
所有评论(0)