小米游戏本基于anaconda安装pytorch(RTX2060+win10)

用的是小米游戏本 RTX2060

基于anaconda,主要参考这一篇:

https://www.cnblogs.com/zhouzhiyao/p/11784055.html

1、进入命令行,查看cuda版本

2、安装cudnn,教程很多,要注册账户,填调查问卷。解压缩,复制粘贴即可,一般不会出大问题。

参考:https://blog.csdn.net/zhangpengzp/article/details/85008507

3、装python3.7

4、装pytorch,用国内的镜像源,不然速度太慢

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1

5、测试

6、安装pychram,用社区版就行

7、在pycharm中选用conda环境

8、在pycharm环境中运行torch程序,我用的是莫烦python-pytorch例程。import torch语句没有报错,大功告成。

9、安装一些库的方法

(1)可以在anaconda navigator中选择直接下载;

(2)可以在anaconda prompt中安装,比如sklearn.

猜你喜欢

转载自blog.csdn.net/cheweng4363/article/details/105720725
今日推荐