1.conda创建虚拟环境pytorch_gpu
conda create -n pytorch_gpu python=3.6
创建虚拟环境还是相对较快的,它会自动为本环境安装一些基本的库,等待时间无需很长,成功之后界面如下所示:
2.切换到pytorch环境
使用如下命令,切换到我们刚刚创建好的pytorch虚拟环境,这样我们避免与其它python环境之间的干扰。
conda activeta pytorch_gpu
切换成功之后就会看到在路径前边显示我们已经进入该虚拟环境。
3.安装几个常用库(也可暂时不安)
conda install pandas jupyter notebook
4.安装pytorch
4.1进入官网查看要下载的版本
查看对应的版本,这里是官方链接:
4.2 根据系 统信息及cuda版本选择对应toolkit
这里最主要的是那个CUDA的版本,此处我选择的是10.1&#x