Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn)
Anaconda+pycharm下配置pytorch-gpu环境(无需单独安装CUDA+Cudnn)安装Anacondapytorch-gpu安装1:打开Anaconda prompt创建虚拟环境conda create -n pytorch python=3.62:创建后激活 环境activate pytorch3:到pytorch官网,根据安装的环境、python和CUDA版本查看指令输入co
·
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)