目录
1.2 pytorch-gpu安装
1.1 Anaconda安装
anaconda的安装请看我之前发的tensoflow-gpu安装,里面有详细的安装过程,这里不做重复描述,传送门
1.2 pytorch-gpu安装
anconda下载完成后,打开anaconda prompt,输入
conda create -n pytorch-gpu python=3.7 创建pytorch用的虚拟环境,当然你们虚拟环境也可以不用命名成pytorch-gpu,创建好后,在pytorch官网找到对应的安装包就行下载,我这里直接
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
如果安装太慢就需要加上镜像源了
永久生效 (pip config set global.index-url+国内镜像地址)
pip config set gl