在此之前默认Anaconda,CUDA, cuDNN 已安装好了。CUDA, cuDNN 下载对应版本安装。
我的环境为:win10+Anaconda+python3.6+cuda10.0
1、打开anaconda prompt
命令行输入:conda create --name pytorch_gpu python=3.6
python_gpu为anaconda下虚拟环境名称,可自定义,python=3.6为选择安装的python版本。
proceed选择y,回车, 等待相关包下载,可以看到在安装完成之后,信息提示:
如果要启用创建的环境,输入指令
conda activate pytorch_gpu
关闭该环境的话,输入指令
conda deactivate
2、安装pytorch
在刚刚创建的环境中安装pytorch, 在