最好创建一个虚拟环境,方便管理。

1.进入Anaconda Prompt 进行创建一个新环境,这边指定python环境是3.6

conda create --name pytorch_gpu python=3.6

 删除命令

conda remove -n 虚拟环境的名称 --all

查看已有的虚拟环境 

conda info -e

2.进入虚拟环境

conda activate pytorch_gpu

3.进行更换下载源,原始源下载速度较慢

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

4.设置搜索时显示通道地址

conda config --set show_channel_urls yes

5.进行下载

进入网站,查看需要下载的版本,进行命令行复制,删除-c pytorch,不然还是会用原始源下载

PyTorchhttps://pytorch.org/

conda install pytorch torchvision torchaudio cudatoolkit=10.2

只安装cpu的

conda install pytorch torchvision torchaudio cpuonly

指定版本

conda install pytorch==版本号 torchvision==版本号 torchaudio==版本号 

6.pycharm配置本地python虚拟环境

       1、进入setting页面

        

        2、进入Python Interpreter页面

        

        3、进行添加环境

        

         

         4.添加虚拟环境下的Python.exe文件即可

        

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐