要点:

1. 修改conda的源,用清华源,具体修改方法详见

anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

2. 创建conda虚拟环境

网上类似教程很多,搜索即可

3. 安装pytorch

官方方法为Start Locally | PyTorch

具体就是运行

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

很多同志运行上面命令会出错,这里解释一下“-c”参数的含义,-c指的是从anaconda官方源的pytorch下载。这个是国外的地址,速度很慢。所以需要进行第四步的修改

4. 修改命令

错误方法:conda install pytorch torchvision torchaudio pytorch-cuda=11.7

本方法直接取消-c参数,但是提示各个包版本不兼容

正确方法:

conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 cudatoolkit=11.7

直接强制指定各个库版本即可。

其中pytorch、torchvision、torchaudio版本有关联关系。如果要安装其他版本也可以,只要关联关系对应就可以。

参考http://t.csdn.cn/17J74

http://t.csdn.cn/vxvsb两篇文章,在此一并感谢

11-13 14:37