Pytorch1.5.1版本安装的方法步骤
查看自己cuda版本,我的cuda是11版本了,所以可以安装11版本以下的任何版本。进入pytorch官网官网网址:2020年11月19号,更新最简单的是直接按官网给的Run this Command命令,直接安装,如下:解释:-c pytorch,意思是从pytorch网站下载,速度感人,有办法的那就方便多了。按照上面图这样选择,安装pytorch有GPU加速的版本,安装命令可以改下,后面加个豆瓣源,这样下载速度快些。或者直接用conda安装,去掉后面的 -c pytorch如果上面方法都下载慢,那就按下面方法来。先进清华源还要下载对应的torchvision===0.6.1,如图下载好就在命令行进入你下载的路径目录里面安装,并输入下面代码进行离线安装。安装完后还要安装cudatoolkit=10.2然后运行测试代码:结果:。可以看出pytorch比TensorFlow1.X好理解,适合人类思维,功能也都全。
用户评论