Anaconda+Cuda+pytorch环境搭建
安装Anaconda
创建conda环境
打开Anaconda Prompt,输入
conda create -n torch python=3.6
torch是环境名称,可自行命名;3.6是python版本。
进入新建的conda环境
conda activate torch
安装Cuda
安装pytorch
官网
查看与Cuda对应的pytorch版本:文章
手动安装对应版本:文章
检验pytorch是否安装完成
进入环境,输入python进入python命令行,输入
import torch
print(torch.version)
print(torch.cuda.is_available())