1)查看系统环境cuda版本说明:对conda熟悉着,可以直接从(4)步开始看。
使用nvcc -V 可以查看系统环境下的cuda版本。
2)查看版本支持nvcc -V
以管理员身份打开Anaconda Prompt,输入下列命令查看conda目前支持的cudatoolkit版本。(注cudatoolkit也是cuda)
conda search cudatoolkit --info
可以看到conda支持cuda10.1,很OK
3)创建虚拟环境
创建一个名为tensorflow的虚拟环境,python为3.7版本。
conda create -n tensorflow python=3.7
然后激活环境,如下图所示。
4)添加源conda activate tensorflow
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
5)获取安装命令
Start Locally | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/
获得命令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
【一定要删除- c pytorch 不然巨特么慢】
conda install pytorch torchvision torchaudio cudatoolkit=11.3
6)测试是否安装成功pytorch安装出现报错
【WARNING conda.gateways.disk.delete:unlink_or_rename_to_trash(140): Could not remove or rename D:\anaconda\pkgs\pytorch-1.5.0-py3.7_cuda102_cudnn7_0.tar.bz2. Please remove this file manually (you may need to reboot to free file handles)
WARNING conda.gateways.disk.delete:unlink_or_rename_to_trash(140): Could not remove or rename D:\anaconda\pkgs\pytorch-1.5.0-py3.7_cuda102_cudnn7_0\Lib\site-packages\torch\lib\cudnn64_7.dll. Please remove this file manually (you may need to reboot to free file handles)】解决
输入:conda install cudnn
下载安装之后再输入安装pytorch的命令即可。
import torch
print(torch.__version__)
print(torch.cuda.is_available())
#True 安装成功
如果不行请进行离线安装!!!!!!!!!!!!!
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)