株洲做网站,解决网站提示有风险,河南和城乡建设厅网站,wordpress 4.8 中文版本文前提#xff1a;
1.你已经下载好了anaconda,最好是新建一个虚拟环境来安装pytorch的GPU版本#xff0c;并且设置了国内镜像源#xff1b; 2.了解自己电脑对应的cuda版本#xff0c;可通过nvidia-smi#xff0c;并下载好了cuda。 安装pytorch的GPU版本
到官网中https…本文前提
1.你已经下载好了anaconda,最好是新建一个虚拟环境来安装pytorch的GPU版本并且设置了国内镜像源 2.了解自己电脑对应的cuda版本可通过nvidia-smi并下载好了cuda。 安装pytorch的GPU版本
到官网中https://pytorch.org/首页向下翻找到以下内容然后根据自己的cuda版本复制相应的命令。安装包相对较大请耐心等待。 安装完成后通过conda list可查看已下载的依赖
gpu版本能够看到cuda相应的依赖包而以下是非gpu版本 你也可以通过查看以下的输出查看是否安装成功
torch.__version__
torch.cuda.is_available()
torch.cuda.get_device_name(0)
